Merge pull request cupy#8897 from kmaehashi/bp-8847-v13-update_pylibr…

…aft_to_cuvs [backport] Updating pylibraft `pairwise_distance` to cuvs
chainer-ci · Jan 28, 2025 · aa5ae4c · aa5ae4c
2 parents 69f990c + d50fc4f
commit aa5ae4c
Show file tree

Hide file tree

Showing 6 changed files with 224 additions and 106 deletions.
diff --git a/.pfnci/linux/tests/cuda-rapids.Dockerfile b/.pfnci/linux/tests/cuda-rapids.Dockerfile
@@ -1,9 +1,11 @@
-ARG BASE_IMAGE="rapidsai/rapidsai-core-dev:23.06-cuda11.8-devel-ubuntu22.04-py3.10"
+ARG BASE_IMAGE="rapidsai/base:25.02a-cuda12.5-py3.11"
 FROM ${BASE_IMAGE}
 
+USER root
+
 RUN export DEBIAN_FRONTEND=noninteractive && \
     apt-get -qqy update && \
-    apt-get -qqy install ccache git curl
+    apt-get -qqy install build-essential cuda-toolkit-12-5 ccache git curl
 
+ENV CUDA_PATH "/usr/local/cuda"
 ENV PATH "/usr/lib/ccache:${PATH}"
-ENV DISABLE_JUPYTER true
diff --git a/.pfnci/linux/tests/cuda-rapids.sh b/.pfnci/linux/tests/cuda-rapids.sh
@@ -14,5 +14,5 @@ export NVCC="ccache nvcc"
 CXX=/usr/lib/ccache/g++ CUDA_PATH="/usr/local/cuda" pip install -v ".[test]"
 "$ACTIONS/unittest.sh" "not slow" \
     "cupyx_tests/scipy_tests/sparse_tests/csgraph_tests" \
-    "cupyx_tests/scipy_tests/spatial_tests"
+    "cupyx_tests/scipy_tests/spatial_tests/test_distance.py"
 "$ACTIONS/cleanup.sh"