Restore disabling benchmarks from ci scripts (removed in #493)

NVIDIA · Feb 29, 2024 · de76435 · de76435
1 parent 75e867e
commit de76435
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/ci/build_common.sh b/ci/build_common.sh
@@ -47,7 +47,7 @@ while [ "${#args[@]}" -ne 0 ]; do
     -std)  CXX_STANDARD="${args[1]}";  args=("${args[@]:2}");;
     -cuda) CUDA_COMPILER="${args[1]}"; args=("${args[@]:2}");;
     -arch) CUDA_ARCHS="${args[1]}";    args=("${args[@]:2}");;
-    -disable-benchmarks) ENABLE_CUB_BENCHMARKS="false"; args=("${args[@]:1}");;
+    -disable-benchmarks) DISABLE_CUB_BENCHMARKS=1; args=("${args[@]:1}");;
     -cmake-options)
         if [ -n "${args[1]}" ]; then
             IFS=' ' read -ra split_args <<< "${args[1]}"

diff --git a/ci/build_cub.sh b/ci/build_cub.sh
@@ -33,6 +33,10 @@ if [[ "$HOST_COMPILER" == *icpc* ]]; then
     ENABLE_CUB_BENCHMARKS="false"
 fi
 
+if [[ $DISABLE_CUB_BENCHMARKS ]]; then
+    ENABLE_CUB_BENCHMARKS="false"
+fi
+
 PRESET="cub-cpp$CXX_STANDARD"
 
 CMAKE_OPTIONS="