Skip to content

Commit 827beed

Browse files
authored
[CMake] Enable cuda lang if USE_CUDA is on (#16426)
1 parent e1c430c commit 827beed

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

cmake/modules/CUDA.cmake

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@ if(USE_CUDA)
2929
message(FATAL_ERROR "Cannot find CUDA, USE_CUDA=" ${USE_CUDA})
3030
endif()
3131
message(STATUS "Build with CUDA ${CUDA_VERSION} support")
32+
enable_language(CUDA)
3233
tvm_file_glob(GLOB RUNTIME_CUDA_SRCS src/runtime/cuda/*.cc)
3334
list(APPEND RUNTIME_SRCS ${RUNTIME_CUDA_SRCS})
3435
list(APPEND COMPILER_SRCS src/target/opt/build_cuda_on.cc)
@@ -62,8 +63,6 @@ if(USE_CUDA)
6263

6364
if(USE_THRUST)
6465
message(STATUS "Build with Thrust support")
65-
cmake_minimum_required(VERSION 3.13) # to compile CUDA code
66-
enable_language(CUDA)
6766
set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} --expt-extended-lambda")
6867
tvm_file_glob(GLOB CONTRIB_THRUST_SRC src/runtime/contrib/thrust/*.cu)
6968
list(APPEND RUNTIME_SRCS ${CONTRIB_THRUST_SRC})
@@ -72,8 +71,6 @@ if(USE_CUDA)
7271
if(USE_CURAND)
7372
message(STATUS "Build with cuRAND support")
7473
message(STATUS "${CUDA_CURAND_LIBRARY}")
75-
cmake_minimum_required(VERSION 3.13) # to compile CUDA code
76-
enable_language(CUDA)
7774
tvm_file_glob(GLOB CONTRIB_CURAND_SRC_CC src/runtime/contrib/curand/*.cc)
7875
tvm_file_glob(GLOB CONTRIB_CURAND_SRC_CU src/runtime/contrib/curand/*.cu)
7976
list(APPEND TVM_RUNTIME_LINKER_LIBS ${CUDA_CURAND_LIBRARY})

0 commit comments

Comments
 (0)