File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed
Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -8,13 +8,14 @@ if(NOT Torch_FOUND)
88 set (LIBTORCH_VERSION "2.9.1" )
99 set (USE_CUDA_LIBTORCH FALSE ) #default is not to use the cuda version but cpu version
1010 if (GAUXC_HAS_CUDA)
11+ find_package (CUDAToolkit)
1112 set (SUPPORTED_CUDA_VERSION_STRINGS "126" "128" "130" )
1213 set (CUDA_VERSION_STRING ${CUDAToolkit_VERSION_MAJOR}${CUDAToolkit_VERSION_MINOR} )
1314 if (CUDA_VERSION_STRING IN_LIST SUPPORTED_CUDA_VERSION_STRINGS)
1415 set (USE_CUDA_LIBTORCH TRUE )
1516 else ()
16- message (WARNING "CUDA toolkit version is ${CUDAToolkit_VERSION} , for which there is no libtorch to download." )
17- message (WARNING "Falling back to cpu version of libtorch." )
17+ message (WARNING "CUDA toolkit version is ${CUDAToolkit_VERSION} , for which there is no libtorch to download." )
18+ message (WARNING "Falling back to cpu version of libtorch." )
1819 endif ()
1920 endif ()
2021
You can’t perform that action at this time.
0 commit comments