lxchen/Xplace_for_ICCAD
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

quick fix to support more CUDA ARCH

Browse Source
This commit is contained in:
liulixinkerry 2023-06-02 17:06:09 +08:00
parent 0aa58ed4a5
commit 1d67a58479
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
CMakeLists.txt
View File

@ -76,7 +76,7 @@ message(STATUS TORCH_ENABLE_CUDA=${TORCH_ENABLE_CUDA})
# set cuda arch and nvcc flags
if (CUDA_FOUND)
if (NOT CUDA_ARCH_LIST)
set(CUDA_ARCH_LIST 8.0 8.6)
set(CUDA_ARCH_LIST 6.0 6.1 7.0 7.5 8.0 8.6)
endif(NOT CUDA_ARCH_LIST)
# for cuda_add_library
cuda_select_nvcc_arch_flags(CUDA_ARCH_FLAGS ${CUDA_ARCH_LIST})