Adds CUDA path for x86-64 (#957)

This commit is contained in:
Ali Hassani 2023-05-24 07:21:25 -07:00 committed by GitHub
parent f079619f5e
commit b4ab501767
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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