Passing USE_CUDA and USE_OPENMP to cmake in build-legate-wheel.

nv-legate · Nov 1, 2023 · 7bd54f0 · 7bd54f0
1 parent 23c523f
commit 7bd54f0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/continuous_integration/scripts/build-legate b/continuous_integration/scripts/build-legate
@@ -2,8 +2,8 @@
 
 . conda-utils;
 
-export CONDA_BUILD_THREADS=1
-export JOBS=1
+# export CONDA_BUILD_THREADS=1
+# export JOBS=1
 
 build_legate_ci() {
     build-legate-cpp;

diff --git a/continuous_integration/scripts/build-legate-wheel b/continuous_integration/scripts/build-legate-wheel
@@ -1,7 +1,7 @@
 #!/usr/bin/env bash
 
 build_legate_wheel() {
-    set -euo pipefail;
+    set -xeuo pipefail;
 
     rm -rf $BASE_DIR/_skbuild;
 
@@ -22,6 +22,9 @@ build_legate_wheel() {
     local cmake_args=(${CMAKE_ARGS:-});
     cmake_args+=("-DFIND_LEGATE_CORE_CPP=ON");
     cmake_args+=("-Dlegate_core_ROOT=$BASE_DIR/legate/build");
+    cmake_args+=("-DLegion_USE_CUDA=${USE_CUDA:-OFF}");
+    cmake_args+=("-DLegion_USE_OpenMP=${USE_OPENMP:-OFF}");
+
 
     # Build + package legate.core Python wheel
     time CMAKE_GENERATOR="Ninja" \