Enable UCX and add test verbosity

nv-legate · Oct 17, 2023 · 6e5e197 · 6e5e197
1 parent 2499f87
commit 6e5e197
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 3 deletions.
diff --git a/continuous_integration/home/coder/.local/bin/build-legate-conda b/continuous_integration/home/coder/.local/bin/build-legate-conda
@@ -45,6 +45,8 @@ python:
   - "${python_version}"
 numpy_version:
   - ">=1.22"
+ucx:
+  - ">=1.14"
 use_local_path:
   - "true"
 gpu_enabled:

diff --git a/continuous_integration/home/coder/.local/bin/build-legate-cpp b/continuous_integration/home/coder/.local/bin/build-legate-cpp
@@ -9,6 +9,7 @@ build_legate_cpp() {
     cmake_args+=(-DBUILD_MARCH=${BUILD_MARCH:-haswell});
     cmake_args+=(-DCMAKE_BUILD_TYPE=Release);
     cmake_args+=(-DLegion_CUDA_ARCH=all-major);
+    cmake_args+=(-DLegion_NETWORKS=ucx);
     cmake_args+=(-DLegion_NETWORKS=);
     cmake_args+=(-DLegion_USE_Python=ON);
     cmake_args+=(-DLegion_PYTHON_EXTRA_INSTALL_ARGS="--root;/;--prefix;\"\${CMAKE_INSTALL_PREFIX}\"");

diff --git a/continuous_integration/home/coder/.local/bin/conda-utils b/continuous_integration/home/coder/.local/bin/conda-utils
@@ -20,7 +20,7 @@ generate_yaml_file() {
             --ctk ${cuda_version}               \
             --python ${python_version}          \
             --openmpi                           \
-            --no-ucx                            \
+            --ucx                               \
         | head -n1 | cut -d' ' -f3              \
     )"
 

diff --git a/continuous_integration/home/coder/.local/bin/run-test-or-analysis b/continuous_integration/home/coder/.local/bin/run-test-or-analysis
@@ -17,9 +17,9 @@ run_test_or_analysis() {
     case "$1" in
         "unit")
             echo "Executing unit tests..."
-            mamba install -y -n "${DEFAULT_CONDA_ENV:-legate}" -c conda-forge pytest pytest-mock ipython jupyter_client
+            mamba install -y -n "${DEFAULT_CONDA_ENV:-legate}" -c conda-forge pytest pytest-mock ipython jupyter_client ucx openmpi
             cd ~/legate/tests/unit
-            pytest
+            pytest -v --verbosity=4
             ;;
         "mypy")
           echo "Executing mypy..."