File tree Expand file tree Collapse file tree 1 file changed +1
-4
lines changed Expand file tree Collapse file tree 1 file changed +1
-4
lines changed Original file line number Diff line number Diff 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} )
You can’t perform that action at this time.
0 commit comments