fix undefined cuPointerGetAttribute symbol error (#3628)

This commit is contained in:
Liumengyuan
2025-08-27 20:24:59 +08:00
committed by GitHub
parent c694fa2879
commit 11803e0907

View File

@@ -519,6 +519,7 @@ elif paddle.is_compiled_with_cuda():
sources=sources, sources=sources,
extra_compile_args={"nvcc": nvcc_compile_args}, extra_compile_args={"nvcc": nvcc_compile_args},
libraries=["cublasLt"], libraries=["cublasLt"],
extra_link_args=["-lcuda"],
), ),
packages=find_packages(where="third_party/DeepGEMM"), packages=find_packages(where="third_party/DeepGEMM"),
package_dir={"": "third_party/DeepGEMM"}, package_dir={"": "third_party/DeepGEMM"},