From 3db985f04ad41245c0651a7b1d4da2c280acb78a Mon Sep 17 00:00:00 2001 From: Anton Volkov Date: Tue, 8 Jul 2025 14:34:20 +0200 Subject: [PATCH] Add a fast path to copy to numpy.ndarray --- dpctl/tensor/_copy_utils.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/dpctl/tensor/_copy_utils.py b/dpctl/tensor/_copy_utils.py index c80733f77f..ecea9f860b 100644 --- a/dpctl/tensor/_copy_utils.py +++ b/dpctl/tensor/_copy_utils.py @@ -41,6 +41,9 @@ def _copy_to_numpy(ary): if not isinstance(ary, dpt.usm_ndarray): raise TypeError(f"Expected dpctl.tensor.usm_ndarray, got {type(ary)}") + if ary.size == 0: + # no data needs to be copied for zero sized array + return np.ndarray(ary.shape, dtype=ary.dtype) nb = ary.usm_data.nbytes q = ary.sycl_queue hh = dpm.MemoryUSMHost(nb, queue=q)