Allow MatDescriptor to be pickle-able

jakirkham · Mar 5, 2020 · 2d2209b · 2d2209b
1 parent 43c3c21
commit 2d2209b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/cupy/cusparse.py b/cupy/cusparse.py
@@ -20,6 +20,9 @@ def create(cls):
         descr = cusparse.createMatDescr()
         return MatDescriptor(descr)
 
+    def __reduce__(self):
+        return self.create, ()
+
     def __del__(self, is_shutting_down=util.is_shutting_down):
         if is_shutting_down():
             return