From 60e06dd70914b9969e815eebfd52677d988b71fd Mon Sep 17 00:00:00 2001 From: Oleksandr Pavlyk Date: Mon, 21 Mar 2022 14:08:51 -0500 Subject: [PATCH] onetrace revealed twice as many copy kernel calls as expected Missing return was the culprit --- dpctl/tensor/_copy_utils.py | 1 + 1 file changed, 1 insertion(+) 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)