mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-05 00:33:03 +08:00
fix undefined cuPointerGetAttribute symbol error (#3628)
This commit is contained in:
@@ -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"},
|
||||||
|
Reference in New Issue
Block a user