make sure cuda lambda support is enabled for cusparse tests

kokkos · Mar 5, 2020 · c97616e · c97616e
1 parent 0479592
commit c97616e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/cmake/kokkoskernels_tpls.cmake b/cmake/kokkoskernels_tpls.cmake
@@ -458,3 +458,12 @@ STRING(REPLACE ";" "\n" KOKKOSKERNELS_TPL_EXPORT_TEMP "${KOKKOSKERNELS_TPL_EXPOR
 #Convert to a regular variable
 UNSET(KOKKOSKERNELS_TPL_EXPORTS CACHE)
 SET(KOKKOSKERNELS_TPL_EXPORTS ${KOKKOSKERNELS_TPL_EXPORT_TEMP})
+
+IF (KOKKOSKERNELS_ENABLE_TPL_CUSPARSE AND KOKKOSKERNELS_ENABLE_TESTS)
+ #The tests use CUDA lambdas, make sure Kokkos was built
+ #with CUDA lambda support
+ KOKKOS_CHECK(OPTIONS CUDA_LAMBDA RETURN_VALUE HAVE_CUDA_LAMBDA)
+ IF (NOT HAVE_CUDA_LAMBDA)
+   MESSAGE(FATAL_ERROR "CUSPARSE tests require Kokkos to be built with CUDA lambda. Please reinstall Kokkos using -DKokkos_ENABLE_CUDA_LAMBDA=ON")
+ ENDIF()
+ENDIF()
diff --git a/scripts/cm_test_all_sandia b/scripts/cm_test_all_sandia
@@ -914,6 +914,7 @@ setup_env() {
     # device tpls
     if [[ "$compiler" == cuda* ]]; then
       NEW_TPL_LIST="cublas,cusparse,"
+      export KOKKOS_CUDA_OPTIONS="${KOKKOS_CUDA_OPTIONS},enable_lambda"
     fi
     # host tpls - use mkl with intel, else use host blas
     if [[ "$compiler" == intel* ]]; then