diff --git a/setup.py b/setup.py index 72db050..2f4a427 100644 --- a/setup.py +++ b/setup.py @@ -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',