update strides

inducer · Sep 30, 2024 · eab5edc · eab5edc
1 parent 498d7e2
commit eab5edc
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/pyopencl/array.py b/pyopencl/array.py
@@ -786,7 +786,13 @@ def __setstate__(self, state):
         else:
             self.base_data = self.allocator(self.nbytes)
 
-        self.set(state["data"], queue=queue)
+        ary = state["data"]
+
+        # Mimics the stride update in _get() below
+        if ary.strides != self.strides:
+            ary = _as_strided(ary, strides=self.strides)
+
+        self.set(ary, queue=queue)
 
     # }}}