Small bugfix in _zero_like

ndgrigorian · ndgrigorian · commit 434015f5bfec · 2023-11-23T16:19:13.000-08:00
_zero_like did not have logic accounting for 0D arrays, so `x.imag` failed for 0D x
diff --git a/dpctl/tensor/_usmarray.pyx b/dpctl/tensor/_usmarray.pyx
@@ -1311,7 +1311,7 @@ cdef usm_ndarray _zero_like(usm_ndarray ary):
     """
     cdef dt = _make_typestr(ary.typenum_)
     cdef usm_ndarray r = usm_ndarray(
-        _make_int_tuple(ary.nd_, ary.shape_),
+        _make_int_tuple(ary.nd_, ary.shape_) if ary.nd_ > 0 else tuple(),
         dtype=dt,
         buffer=ary.base_.get_usm_type()
     )

Original file line number	Diff line number	Diff line change
`@@ -1311,7 +1311,7 @@ cdef usm_ndarray _zero_like(usm_ndarray ary):`
`1311`	`1311`	`"""`
`1312`	`1312`	`cdef dt = _make_typestr(ary.typenum_)`
`1313`	`1313`	`cdef usm_ndarray r = usm_ndarray(`
`1314`		`- _make_int_tuple(ary.nd_, ary.shape_),`
	`1314`	`+ _make_int_tuple(ary.nd_, ary.shape_) if ary.nd_ > 0 else tuple(),`
`1315`	`1315`	`dtype=dt,`
`1316`	`1316`	`buffer=ary.base_.get_usm_type()`
`1317`	`1317`	`)`