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

Adds CUDA path for x86-64 (#957)

Browse Source
This commit is contained in:
Ali Hassani
2023-05-24 07:21:25 -07:00
committed by GitHub
parent f079619f5e
commit b4ab501767
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
CUDA.cmake
View File

@ -76,6 +76,7 @@ find_library(
PATHS
${CUDA_TOOLKIT_ROOT_DIR}
PATH_SUFFIXES
lib/x86_64-linux-gnu
lib/x64
lib64
lib
@ -120,6 +121,7 @@ find_library(
PATHS
${CUDA_TOOLKIT_ROOT_DIR}
PATH_SUFFIXES
lib/x86_64-linux-gnu
lib/x64
lib64
lib