Use properly imported MatDescriptor for cupy dispatch registr…

…ation (#7868)
dask · May 30, 2023 · 3286d2f · 3286d2f
1 parent 61efdfe
commit 3286d2f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/distributed/protocol/cupy.py b/distributed/protocol/cupy.py
@@ -85,7 +85,7 @@ def dask_deserialize_cupy_ndarray(header, frames):
     def reduce_matdescriptor(other):
         # Pickling MatDescriptor errors
         # xref: https://github.com/cupy/cupy/issues/3061
-        return cupy.cusparse.MatDescriptor.create, ()
+        return MatDescriptor.create, ()
 
     copyreg.pickle(MatDescriptor, reduce_matdescriptor)