Merge branch 'branch-23.12' into enh_cagra_bench_ivf_pq_config

ci/test_cpp.sh

@@ -36,6 +36,7 @@ trap "EXITCODE=1" ERR
 set +e
 
 # Run libraft gtests from libraft-tests package
+cd "$CONDA_PREFIX"/bin/gtests/libraft
 ctest -j8 --output-on-failure
 
 rapids-logger "Test script exiting with value: $EXITCODE"

conda/environments/all_cuda-118_arch-aarch64.yaml

@@ -0,0 +1,62 @@
+# This file is generated by `rapids-dependency-file-generator`.
+# To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.
+channels:
+- rapidsai
+- rapidsai-nightly
+- dask/label/dev
+- conda-forge
+- nvidia
+dependencies:
+- breathe
+- c-compiler
+- clang-tools=16.0.6
+- clang==16.0.6
+- cmake>=3.26.4
+- cuda-profiler-api=11.8.86
+- cuda-python>=11.7.1,<12.0a0
+- cuda-version=11.8
+- cudatoolkit
+- cupy>=12.0.0
+- cxx-compiler
+- cython>=3.0.0
+- dask-core==2023.9.2
+- dask-cuda==23.12.*
+- dask==2023.9.2
+- distributed==2023.9.2
+- doxygen>=1.8.20
+- gcc_linux-aarch64=11.*
+- gmock>=1.13.0
+- graphviz
+- gtest>=1.13.0
+- ipython
+- joblib>=0.11
+- libcublas-dev=11.11.3.6
+- libcublas=11.11.3.6
+- libcurand-dev=10.3.0.86
+- libcurand=10.3.0.86
+- libcusolver-dev=11.4.1.48
+- libcusolver=11.4.1.48
+- libcusparse-dev=11.7.5.86
+- libcusparse=11.7.5.86
+- nccl>=2.9.9
+- ninja
+- numba>=0.57
+- numpy>=1.21
+- numpydoc
+- nvcc_linux-aarch64=11.8
+- pre-commit
+- pydata-sphinx-theme
+- pytest
+- pytest-cov
+- recommonmark
+- rmm==23.12.*
+- scikit-build>=0.13.1
+- scikit-learn
+- scipy
+- sphinx-copybutton
+- sphinx-markdown-tables
+- sysroot_linux-aarch64==2.17
+- ucx-proc=*=gpu
+- ucx-py==0.35.*
+- ucx>=1.13.0
+name: all_cuda-118_arch-aarch64

conda/environments/all_cuda-120_arch-aarch64.yaml

@@ -0,0 +1,58 @@
+# This file is generated by `rapids-dependency-file-generator`.
+# To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.
+channels:
+- rapidsai
+- rapidsai-nightly
+- dask/label/dev
+- conda-forge
+- nvidia
+dependencies:
+- breathe
+- c-compiler
+- clang-tools=16.0.6
+- clang==16.0.6
+- cmake>=3.26.4
+- cuda-cudart-dev
+- cuda-nvcc
+- cuda-profiler-api
+- cuda-python>=12.0,<13.0a0
+- cuda-version=12.0
+- cupy>=12.0.0
+- cxx-compiler
+- cython>=3.0.0
+- dask-core==2023.9.2
+- dask-cuda==23.12.*
+- dask==2023.9.2
+- distributed==2023.9.2
+- doxygen>=1.8.20
+- gcc_linux-aarch64=11.*
+- gmock>=1.13.0
+- graphviz
+- gtest>=1.13.0
+- ipython
+- joblib>=0.11
+- libcublas-dev
+- libcurand-dev
+- libcusolver-dev
+- libcusparse-dev
+- nccl>=2.9.9
+- ninja
+- numba>=0.57
+- numpy>=1.21
+- numpydoc
+- pre-commit
+- pydata-sphinx-theme
+- pytest
+- pytest-cov
+- recommonmark
+- rmm==23.12.*
+- scikit-build>=0.13.1
+- scikit-learn
+- scipy
+- sphinx-copybutton
+- sphinx-markdown-tables
+- sysroot_linux-aarch64==2.17
+- ucx-proc=*=gpu
+- ucx-py==0.35.*
+- ucx>=1.13.0
+name: all_cuda-120_arch-aarch64

conda/environments/bench_ann_cuda-118_arch-aarch64.yaml

@@ -0,0 +1,43 @@
+# This file is generated by `rapids-dependency-file-generator`.
+# To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.
+channels:
+- rapidsai
+- rapidsai-nightly
+- dask/label/dev
+- conda-forge
+- nvidia
+dependencies:
+- benchmark>=1.8.2
+- c-compiler
+- clang-tools=16.0.6
+- clang==16.0.6
+- cmake>=3.26.4
+- cuda-profiler-api=11.8.86
+- cuda-version=11.8
+- cudatoolkit
+- cxx-compiler
+- cython>=3.0.0
+- gcc_linux-aarch64=11.*
+- glog>=0.6.0
+- h5py>=3.8.0
+- hnswlib=0.7.0
+- libcublas-dev=11.11.3.6
+- libcublas=11.11.3.6
+- libcurand-dev=10.3.0.86
+- libcurand=10.3.0.86
+- libcusolver-dev=11.4.1.48
+- libcusolver=11.4.1.48
+- libcusparse-dev=11.7.5.86
+- libcusparse=11.7.5.86
+- matplotlib
+- nccl>=2.9.9
+- ninja
+- nlohmann_json>=3.11.2
+- nvcc_linux-aarch64=11.8
+- openblas
+- pandas
+- pyyaml
+- rmm==23.12.*
+- scikit-build>=0.13.1
+- sysroot_linux-aarch64==2.17
+name: bench_ann_cuda-118_arch-aarch64

conda/recipes/raft-dask/conda_build_config.yaml

@@ -14,7 +14,7 @@ sysroot_version:
   - "2.17"
 
 ucx_version:
-  - ">=1.13.0,<1.15.0"
+  - ">=1.14.1,<1.16.0"
 
 ucx_py_version:
   - "0.35.*"

cpp/bench/prims/core/bitset.cu

@@ -44,7 +44,7 @@ struct bitset_bench : public fixture {
     loop_on_state(state, [this]() {
       auto my_bitset = raft::core::bitset<bitset_t, index_t>(
         this->res, raft::make_const_mdspan(mask.view()), params.bitset_len);
-      my_bitset.test(res, raft::make_const_mdspan(queries.view()), outputs.view());
+      my_bitset.test(this->res, raft::make_const_mdspan(queries.view()), outputs.view());
     });
   }