Removes malformed fast path for size-zero arrays in repeat

ndgrigorian · ndgrigorian · commit 57033b0e8277 · 2024-05-16T11:49:02.000-07:00
diff --git a/dpctl/tensor/_manipulation_functions.py b/dpctl/tensor/_manipulation_functions.py
@@ -838,9 +838,6 @@ def repeat(x, repeats, /, *, axis=None):
             f"got {type(repeats)}"
         )
 
-    if axis_size == 0:
-        return dpt.empty(x_shape, dtype=x.dtype, sycl_queue=exec_q)
-
     if scalar:
         res_axis_size = repeats * axis_size
         if axis is not None:

Original file line number	Diff line number	Diff line change
`@@ -838,9 +838,6 @@ def repeat(x, repeats, /, *, axis=None):`
`838`	`838`	`f"got {type(repeats)}"`
`839`	`839`	`)`
`840`	`840`
`841`		`- if axis_size == 0:`
`842`		`- return dpt.empty(x_shape, dtype=x.dtype, sycl_queue=exec_q)`
`843`		`-`
`844`	`841`	`if scalar:`
`845`	`842`	`res_axis_size = repeats * axis_size`
`846`	`843`	`if axis is not None:`