Skip to content

Commit b1b7be5

Browse files
Fix Cuda compilation (#317)
1 parent c4eb386 commit b1b7be5

File tree

1 file changed

+2
-8
lines changed

1 file changed

+2
-8
lines changed

ucm/sparse/kvstar/retrieve/CMakeLists.txt

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,8 @@
11
# auto detect cuda ------------------------
22
if($ENV{PLATFORM} STREQUAL "cuda")
33
if(NOT DEFINED CMAKE_CUDA_COMPILER)
4-
set(CUDA_HOME "$ENV{CUDA_HOME}")
5-
if(NOT CUDA_HOME)
6-
set(CUDA_HOME "/usr/local/cuda")
7-
endif()
8-
if(NOT EXISTS "${CUDA_HOME}")
9-
message(FATAL_ERROR "CUDA_HOME directory does not exist: ${CUDA_HOME}")
10-
endif()
11-
set(CMAKE_CUDA_COMPILER "${CUDA_HOME}/bin/nvcc" CACHE FILEPATH "CUDA compiler" FORCE)
4+
set(CUDA_ROOT "/usr/local/cuda/" CACHE PATH "Path to CUDA root directory")
5+
set(CMAKE_CUDA_COMPILER ${CUDA_ROOT}/bin/nvcc)
126
endif()
137
enable_language(CUDA)
148
set(CMAKE_CUDA_ARCHITECTURES 75 80 86 89 90)

0 commit comments

Comments
 (0)