reshaped_strides is also called from shape setter

Special case setting shape for zero-element arrays

Author: oleksandr-pavlyk

dpctl/tensor/_usmarray.pyx

@@ -428,6 +428,7 @@ cdef class usm_ndarray:
         cdef int contig_flag = 0
         cdef Py_ssize_t *shape_ptr = NULL
         cdef Py_ssize_t *strides_ptr = NULL
+        cdef Py_ssize_t size = -1
         import operator
 
         from ._reshape import reshaped_strides
@@ -439,15 +440,19 @@ cdef class usm_ndarray:
             raise TypeError(
                 "Target shape must be a finite iterable of integers"
             )
-        if not np.prod(new_shape) == shape_to_elem_count(self.nd_, self.shape_):
+        size = shape_to_elem_count(self.nd_, self.shape_)
+        if not np.prod(new_shape) == size:
             raise TypeError(
                 f"Can not reshape array of size {self.size} into {new_shape}"
             )
-        new_strides = reshaped_strides(
-            self.shape,
-            self.strides,
-            new_shape
-        )
+        if size > 0:
+            new_strides = reshaped_strides(
+               self.shape,
+               self.strides,
+               new_shape
+            )
+        else:
+            new_strides = (1,) * len(new_shape)
         if new_strides is None:
             raise AttributeError(
                 "Incompatible shape for in-place modification. "

dpctl/tests/test_usm_ndarray_ctor.py

@@ -713,6 +713,21 @@ def relaxed_strides_equal(st1, st2, sh):
     X = dpt.usm_ndarray((4, 4), dtype="d")[::2, ::2]
     with pytest.raises(AttributeError):
         X.shape = (4,)
+    X = dpt.usm_ndarray((0,), dtype="i4")
+    X.shape = (0,)
+    X.shape = (
+        2,
+        0,
+    )
+    X.shape = (
+        0,
+        2,
+    )
+    X.shape = (
+        1,
+        0,
+        1,
+    )
 
 
 def test_len():