biondizzle/DeepGEMM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update setup.py (#134)

Browse Source
This commit is contained in:
dan_the_3rd
2025-07-28 10:58:23 +02:00
committed by GitHub
parent 187656694f
commit 8987798502
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
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',