fix cu118 link issue (#3421)

This commit is contained in:
Yineng Zhang
2025-02-09 18:16:44 +08:00
committed by GitHub
parent 6702592d0e
commit 29daf498cd

View File

@@ -63,7 +63,6 @@ include_dirs = [
flashinfer.resolve() / "include" / "gemm",
flashinfer.resolve() / "csrc",
"cublas",
"cublasLt",
turbomind.resolve(),
turbomind.resolve() / "src",
]
@@ -143,7 +142,7 @@ for flag in [
pass
cxx_flags = ["-O3"]
libraries = ["c10", "torch", "torch_python", "cuda", "cublas", "cublasLt"]
libraries = ["c10", "torch", "torch_python", "cuda", "cublas"]
extra_link_args = ["-Wl,-rpath,$ORIGIN/../../torch/lib", "-L/usr/lib/x86_64-linux-gnu"]
ext_modules = [