[BUG] dask_cudf loc operation fails with cudf Series and cupy array fails

[VibhuJawa]

Describe the bug
dask_cudf loc operation fails with cudf Series and cupy array

Steps/Code to reproduce bug

import dask_cudf
import cudf
import pandas as pd
import numpy as np


df = cudf.DataFrame({'a':np.arange(0,100, dtype=np.int32),
                     'b':np.arange(0,100, dtype=np.int32)
                    })
df = dask_cudf.from_cudf(df,10)
df = df.set_index('b')

df.loc[cudf.Series([0,10,50], dtype=np.int32)]

---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
Cell In [45], line 1
----> 1 df.loc[ar]

File /datasets/vjawa/miniconda3/envs/cugraph_dgl_dev_oct_5/lib/python3.9/site-packages/dask/dataframe/indexing.py:103, in _LocIndexer.__getitem__(self, key)
    101     iindexer = key
    102     cindexer = None
--> 103 return self._loc(iindexer, cindexer)

File /datasets/vjawa/miniconda3/envs/cugraph_dgl_dev_oct_5/lib/python3.9/site-packages/dask/dataframe/indexing.py:125, in _LocIndexer._loc(self, iindexer, cindexer)
    122         return self._loc_list(iindexer.values, cindexer)
    123     else:
    124         # element should raise KeyError
--> 125         return self._loc_element(iindexer, cindexer)
    126 else:
    127     if isinstance(iindexer, (list, np.ndarray)) or (
    128         is_series_like(iindexer) and not is_bool_dtype(iindexer.dtype)
    129     ):
    130         # applying map_partitions to each partition
    131         # results in duplicated NaN rows

File /datasets/vjawa/miniconda3/envs/cugraph_dgl_dev_oct_5/lib/python3.9/site-packages/dask/dataframe/indexing.py:193, in _LocIndexer._loc_element(self, iindexer, cindexer)
    191 def _loc_element(self, iindexer, cindexer):
    192     name = "loc-%s" % tokenize(iindexer, self.obj)
--> 193     part = self._get_partitions(iindexer)
    195     if iindexer < self.obj.divisions[0] or iindexer > self.obj.divisions[-1]:
    196         raise KeyError("the label [%s] is not in the index" % str(iindexer))

File /datasets/vjawa/miniconda3/envs/cugraph_dgl_dev_oct_5/lib/python3.9/site-packages/dask/dataframe/indexing.py:216, in _LocIndexer._get_partitions(self, keys)
    213     return _partitions_of_index_values(self.obj.divisions, keys)
    214 else:
    215     # element
--> 216     return _partition_of_index_value(self.obj.divisions, keys)

File /datasets/vjawa/miniconda3/envs/cugraph_dgl_dev_oct_5/lib/python3.9/site-packages/dask/dataframe/indexing.py:327, in _partition_of_index_value(divisions, val)
    325     raise ValueError(msg)
    326 val = _coerce_loc_index(divisions, val)
--> 327 i = bisect.bisect_right(divisions, val)
    328 return min(len(divisions) - 2, max(0, i - 1))

File cupy/_core/core.pyx:1238, in cupy._core.core._ndarray_base.__nonzero__()

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

Expected behavior
I would expect this will work as it does for numpy or padas series . See below.

df.loc[np.asarray([0,10,50], dtype=np.int32)].compute()
df.loc[pd.Series([0,10,50], dtype=np.int32)].compute()

	a
b	
0	0
10	100
50	500

Additional context
There is no workaround apart from moving our stuff to pandas, numpy array or dask_cudf Series.

[quasiben]

resolved by dask/dask#9634