fix undefined cuPointerGetAttribute symbol error (#3628)

2025-10-04 16:22:57 +08:00 · 2025-08-27 20:24:59 +08:00
parent c694fa2879
commit 11803e0907
1 changed files with 1 additions and 0 deletions
--- a/custom_ops/setup_ops.py
+++ b/custom_ops/setup_ops.py
@@ -519,6 +519,7 @@ elif paddle.is_compiled_with_cuda():
            sources=sources,
            extra_compile_args={"nvcc": nvcc_compile_args},
            libraries=["cublasLt"],
+            extra_link_args=["-lcuda"],
        ),
        packages=find_packages(where="third_party/DeepGEMM"),
        package_dir={"": "third_party/DeepGEMM"},