squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

add library_dirs (#977)

Browse Source
This commit is contained in:
q.yao 2023-06-15 00:09:12 +08:00 committed by GitHub
parent 473a67073e
commit f6d42f2dd0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
python/setup.py
View File

@ -108,10 +108,15 @@ include_dirs = [
] ]
library_dirs = [
cuda_install_path + '/lib64',
]
ext_modules = [ ext_modules = [
Pybind11Extension('cutlass_bindings', Pybind11Extension('cutlass_bindings',
['cutlass/cpp/cutlass_bindings.cpp'], ['cutlass/cpp/cutlass_bindings.cpp'],
include_dirs=include_dirs, include_dirs=include_dirs,
library_dirs=library_dirs,
extra_compile_args=['-Xcompiler="-fpermissive"', '-w', '-std=c++17'], extra_compile_args=['-Xcompiler="-fpermissive"', '-w', '-std=c++17'],
libraries=['cudart']) libraries=['cudart'])
] ]