Skip to content

Commit eab5edc

Browse files
update strides
1 parent 498d7e2 commit eab5edc

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

pyopencl/array.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -786,7 +786,13 @@ def __setstate__(self, state):
786786
else:
787787
self.base_data = self.allocator(self.nbytes)
788788

789-
self.set(state["data"], queue=queue)
789+
ary = state["data"]
790+
791+
# Mimics the stride update in _get() below
792+
if ary.strides != self.strides:
793+
ary = _as_strided(ary, strides=self.strides)
794+
795+
self.set(ary, queue=queue)
790796

791797
# }}}
792798

0 commit comments

Comments
 (0)