From bad9791f8d566a4afeff02a9e0638573da592d2b Mon Sep 17 00:00:00 2001 From: Tarang Jain Date: Fri, 18 Oct 2024 14:14:21 -0700 Subject: [PATCH 1/2] use numpy<2 in github action --- .github/actions/build_cmake/action.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/actions/build_cmake/action.yml b/.github/actions/build_cmake/action.yml index ad3811f473..6d5f68205d 100644 --- a/.github/actions/build_cmake/action.yml +++ b/.github/actions/build_cmake/action.yml @@ -32,7 +32,7 @@ runs: conda update -y -q conda echo "$CONDA/bin" >> $GITHUB_PATH - conda install -y -q python=3.11 cmake make swig numpy scipy pytest gflags + conda install -y -q python=3.11 cmake make swig numpy=1.26.4 scipy pytest gflags # install base packages for ARM64 if [ "${{ runner.arch }}" = "ARM64" ]; then From d8fea6483f1fac6db45de0b0e6b44fb55ab55e5a Mon Sep 17 00:00:00 2001 From: Tarang Jain Date: Fri, 18 Oct 2024 16:13:16 -0700 Subject: [PATCH 2/2] pin numpy for only raft build --- .github/actions/build_cmake/action.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/actions/build_cmake/action.yml b/.github/actions/build_cmake/action.yml index 6d5f68205d..a040cc6a40 100644 --- a/.github/actions/build_cmake/action.yml +++ b/.github/actions/build_cmake/action.yml @@ -32,7 +32,7 @@ runs: conda update -y -q conda echo "$CONDA/bin" >> $GITHUB_PATH - conda install -y -q python=3.11 cmake make swig numpy=1.26.4 scipy pytest gflags + conda install -y -q python=3.11 cmake make swig numpy scipy pytest gflags # install base packages for ARM64 if [ "${{ runner.arch }}" = "ARM64" ]; then @@ -54,7 +54,7 @@ runs: conda install -y -q cuda-toolkit -c "nvidia/label/cuda-12.4.0" # and CUDA from RAFT channel for RAFT builds elif [ "${{ inputs.raft }}" = "ON" ]; then - conda install -y -q libraft=24.06 cuda-version=12.4 cuda-toolkit -c rapidsai -c "nvidia/label/cuda-12.4.0" -c conda-forge + conda install -y -q numpy=1.26.4 libraft=24.06 cuda-version=12.4 cuda-toolkit -c rapidsai -c "nvidia/label/cuda-12.4.0" -c conda-forge fi # install test packages