Merge pull request cupy#8970 from jakirkham/use_rapids_25.02

Fix `cupyx.scipy.spatial.distance`'s `cdist` for RAPIDS 24.12 compatibility
chainer-ci · Feb 22, 2025 · d7313aa · d7313aa
1 parent f5e36bf
commit d7313aa
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/.pfnci/linux/tests/cuda-rapids.Dockerfile b/.pfnci/linux/tests/cuda-rapids.Dockerfile
@@ -1,11 +1,11 @@
-ARG BASE_IMAGE="rapidsai/base:25.02a-cuda12.5-py3.11"
+ARG BASE_IMAGE="rapidsai/base:25.02-cuda12.8-py3.11"
 FROM ${BASE_IMAGE}
 
 USER root
 
 RUN export DEBIAN_FRONTEND=noninteractive && \
     apt-get -qqy update && \
-    apt-get -qqy install build-essential cuda-toolkit-12-5 ccache git curl
+    apt-get -qqy install build-essential cuda-toolkit-12-8 ccache git curl
 
 ENV CUDA_PATH "/usr/local/cuda"
 ENV PATH "/usr/lib/ccache:${PATH}"
diff --git a/cupyx/scipy/spatial/distance.py b/cupyx/scipy/spatial/distance.py
@@ -628,7 +628,7 @@ def cdist(XA, XB, metric='euclidean', out=None, **kwargs):
             output_arr = out if out is not None else cupy.zeros((mA, mB),
                                                                 dtype=XA.dtype,
                                                                 order=XA_order)
-            pairwise_distance(XA, XB, output_arr, metric, p=p)
+            pairwise_distance(XA, XB, output_arr, metric, p)
             return output_arr
         else:
             raise ValueError('Unknown Distance Metric: %s' % mstr)