Skip to content

Commit 0cd0d5e

Browse files
Fix gh-1913
Check array_mask_nelems and early exit if zero.
1 parent 4a94ef9 commit 0cd0d5e

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

dpctl/tensor/_copy_utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -924,6 +924,8 @@ def _place_impl(ary, ary_mask, vals, axis=0):
924924
else:
925925
rhs = dpt.astype(vals, ary.dtype)
926926
rhs = dpt.broadcast_to(rhs, expected_vals_shape)
927+
if mask_nelems == 0:
928+
return
927929
dep_ev = _manager.submitted_events
928930
hev, pl_ev = ti._place(
929931
dst=ary,

0 commit comments

Comments
 (0)