Merge pull request #1080 from IntelPython/fix_meshgrid_empty

Fixed meshgrid() function with empty input.
IntelPython · Feb 22, 2023 · bdeb37d · bdeb37d
2 parents eadc0bd + 2fb6b6c
commit bdeb37d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git 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
@@ -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)))