Bug in cupyx.scipy.spatial.distance.euclidean function

[tung491]

Description

The function cupyx.scipy.spatial.distance.euclidean return the IndexError: tuple index out of range error when pass two 1D array into. The scipy.spatial.distance.euclidean work properly with this input.

---------------------------------------------------------------------------
IndexError                                Traceback (most recent call last)
Cell In [7], line 1
----> 1 distance.euclidean(np.array([83.80576723, 57.99175829]), np.array([50, 50]))

File /usr/local/lib/python3.9/dist-packages/cupyx/scipy/spatial/distance.py:311, in euclidean(u, v)
    309 v = cupy.asarray(v)
    310 output_arr = cupy.zeros((1,), dtype=u.dtype)
--> 311 pairwise_distance(u, v, output_arr, "euclidean")
    313 return output_arr[0]

File handle.pyx:225, in pylibraft.common.handle.auto_sync_handle.wrapper()

File /usr/local/lib/python3.9/dist-packages/pylibraft/common/outputs.py:83, in auto_convert_output.<locals>.wrapper(*args, **kwargs)
     81 @functools.wraps(f)
     82 def wrapper(*args, **kwargs):
---> 83     ret_value = f(*args, **kwargs)
     84     if isinstance(ret_value, pylibraft.common.device_ndarray):
     85         return convert_to_cai_type(ret_value)

File pairwise_distance.pyx:185, in pylibraft.distance.pairwise_distance.distance()

IndexError: tuple index out of range

To Reproduce

from scipy.spatial import distance
distance.euclidean([83.80576723, 57.99175829], [50, 50])

from cupyx.scipy.spatial import distance
import cupy as np
distance.euclidean([83.80576723, 57.99175829], [50, 50])
distance.euclidean(np.array([83.80576723, 57.99175829]), np.array([50, 50]))

Installation

Wheel (pip install cupy-***)

Environment

CUDA Runtime Version         : 11080 (linked to CuPy) / 11060 (locally installed)
cuBLAS Version               : (available)
cuFFT Version                : 10702
cuRAND Version               : 10209
cuSOLVER Version             : (11, 3, 4)
cuSPARSE Version             : (available)
NVRTC Version                : (11, 6)
Thrust Version               : 200200
CUB Build Version            : 200200
Jitify Build Version         : <unknown>
cuDNN Build Version          : 8801
cuDNN Version                : 8302
NCCL Build Version           : None
NCCL Runtime Version         : None
cuTENSOR Version             : None
cuSPARSELt Build Version     : None
Device 0 Name                : Quadro P5000
Device 0 Compute Capability  : 61
Device 0 PCI Bus ID          : 0000:00:05.0

Additional Information

No response

[kmaehashi]

Thanks for reporting! It looks the error is coming from pylibraft? cc/ @cjnolet