del update_inputs.cu (#10022)

This commit is contained in:
tianshuo78520a
2025-03-07 16:54:58 +08:00
committed by GitHub
parent 19e10ba59e
commit 63d920b565

View File

@@ -12,15 +12,19 @@
# See the License for the specific language governing permissions and
# limitations under the License.
from paddle.utils.cpp_extension import CUDAExtension, setup
import subprocess
from paddle.utils.cpp_extension import CUDAExtension, setup
def update_git_submodule():
try:
subprocess.run(["git", "submodule", "update", "--init"], check=True)
except subprocess.CalledProcessError as e:
print(f"Error occurred while updating git submodule: {str(e)}")
raise
update_git_submodule()
setup(
name="paddlenlp_ops",
@@ -42,7 +46,6 @@ setup(
"./gpu/rebuild_padding_v2.cu",
"./gpu/set_value_by_flags_v2.cu",
"./gpu/stop_generation_multi_ends_v2.cu",
"./gpu/update_inputs.cu",
"./gpu/get_output.cc",
"./gpu/save_with_output_msg.cc",
"./gpu/write_int8_cache_kv.cu",