Cmake cuda flags as GPU build is failing

neo-ai · Jan 21, 2021 · 004cf83 · 004cf83
1 parent 4de6d1b
commit 004cf83
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -173,6 +173,7 @@ if(USE_CUDA)
       set(CMAKE_CUDA_STANDARD_REQUIRED ON)
     endif()
     enable_language(CUDA)
+    set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} --expt-extended-lambda")
     file(GLOB CONTRIB_THRUST_SRC ${TVM_SRC}/src/runtime/contrib/thrust/*.cu)
     list(APPEND RUNTIME_CUDA_SRCS ${CONTRIB_THRUST_SRC})