Update setup.py (#134)

This commit is contained in:
dan_the_3rd
2025-07-28 16:58:23 +08:00
committed by GitHub
parent 187656694f
commit 8987798502
+1 -1
View File
@@ -18,7 +18,7 @@ build_include_dirs = [
build_libraries = ['cuda', 'cudart']
build_library_dirs = [
f'{CUDA_HOME}/lib64',
f'{CUDA_HOME}/lib64/stub'
f'{CUDA_HOME}/lib64/stubs'
]
third_party_include_dirs = [
'third-party/cutlass/include/cute',