diff --git a/dpctl/tensor/_copy_utils.py b/dpctl/tensor/_copy_utils.py index 580df3007a..12b0619b6d 100644 --- a/dpctl/tensor/_copy_utils.py +++ b/dpctl/tensor/_copy_utils.py @@ -201,6 +201,7 @@ def _copy_from_usm_ndarray_to_usm_ndarray(dst, src): if dst.ndim == src.ndim and dst.shape == src.shape: _copy_same_shape(dst, src) + return try: common_shape = _broadcast_shapes(dst.shape, src.shape)