From 2fb6b6c45741837b557e54fddf6980facc998363 Mon Sep 17 00:00:00 2001 From: Natalia Polina Date: Tue, 21 Feb 2023 14:04:26 -0600 Subject: [PATCH] Fixed meshgrid() function with empty input. --- dpctl/tensor/_ctors.py | 3 +++ dpctl/tests/test_usm_ndarray_ctor.py | 1 + 2 files changed, 4 insertions(+) diff --git a/dpctl/tensor/_ctors.py b/dpctl/tensor/_ctors.py index 2de9c979bb..16388bda5b 100644 --- a/dpctl/tensor/_ctors.py +++ b/dpctl/tensor/_ctors.py @@ -1375,6 +1375,9 @@ def meshgrid(*arrays, indexing="xy"): "Unrecognized indexing keyword value, expecting 'xy' or 'ij.'" ) n = len(arrays) + if n == 0: + return [] + sh = (-1,) + (1,) * (n - 1) res = [] diff --git a/dpctl/tests/test_usm_ndarray_ctor.py b/dpctl/tests/test_usm_ndarray_ctor.py index 946ff53b27..54ccc85916 100644 --- a/dpctl/tests/test_usm_ndarray_ctor.py +++ b/dpctl/tests/test_usm_ndarray_ctor.py @@ -1735,6 +1735,7 @@ def test_meshgrid(): assert n == len(Znp) for i in range(n): assert np.array_equal(dpt.asnumpy(Z[i]), Znp[i]) + assert dpt.meshgrid() == [] # dimension > 1 must raise ValueError with pytest.raises(ValueError): dpt.meshgrid(dpt.usm_ndarray((4, 4)))