From c9ea41346b60b22e496678be561d58025275e50f Mon Sep 17 00:00:00 2001
From: Charles Blackmon-Luca <20627856+charlesbluca@users.noreply.github.com>
Date: Tue, 30 May 2023 09:26:47 -0700
Subject: [PATCH] Use properly imported MatDescriptor for dispatch registration

---
 distributed/protocol/cupy.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/distributed/protocol/cupy.py b/distributed/protocol/cupy.py
index fc6e0d08712..2cc2169b071 100644
--- 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)