Adds tests for #1455 resolution

Author: ndgrigorian

dpctl/tests/test_tensor_sum.py

@@ -212,6 +212,24 @@ def test_axis0_bug():
     assert dpt.all(s == expected)
 
 
+def test_sum_axis1_axis0():
+    """See gh-1455"""
+    get_queue_or_skip()
+
+    # The atomic case is checked in `test_usm_ndarray_reductions`
+    # This test checks the tree reduction path for correctness
+    x = dpt.reshape(dpt.arange(4 * 5, dtype="f4"), (4, 5))
+
+    m = dpt.sum(x, axis=0)
+    expected = dpt.asarray([30, 34, 38, 42, 46], dtype="f4")
+    tol = dpt.finfo(m.dtype).resolution
+    assert dpt.allclose(m, expected, atol=tol, rtol=tol)
+
+    m = dpt.sum(x, axis=1)
+    expected = dpt.asarray([10, 35, 60, 85], dtype="f4")
+    assert dpt.allclose(m, expected, atol=tol, rtol=tol)
+
+
 def _any_complex(dtypes):
     return any(dpt.isdtype(dpt.dtype(dt), "complex floating") for dt in dtypes)
 

dpctl/tests/test_usm_ndarray_reductions.py

@@ -61,6 +61,19 @@ def test_max_min_axis():
     assert dpt.all(m == x[:, 0, 0, :, 0])
 
 
+def test_max_axis1_axis0():
+    """See gh-1455"""
+    get_queue_or_skip()
+
+    x = dpt.reshape(dpt.arange(4 * 5), (4, 5))
+
+    m = dpt.max(x, axis=0)
+    assert dpt.all(m == x[-1, :])
+
+    m = dpt.max(x, axis=1)
+    assert dpt.all(m == x[:, -1])
+
+
 def test_reduction_keepdims():
     get_queue_or_skip()
 
@@ -440,3 +453,27 @@ def test_hypot_complex():
     x = dpt.zeros(1, dtype="c8")
     with pytest.raises(TypeError):
         dpt.reduce_hypot(x)
+
+
+def test_tree_reduction_axis1_axis0():
+    """See gh-1455"""
+    get_queue_or_skip()
+
+    x = dpt.reshape(dpt.arange(4 * 5, dtype="f4"), (4, 5))
+
+    m = dpt.logsumexp(x, axis=0)
+    tol = dpt.finfo(m.dtype).resolution
+    assert_allclose(
+        dpt.asnumpy(m),
+        np.logaddexp.reduce(dpt.asnumpy(x), axis=0, dtype=m.dtype),
+        rtol=tol,
+        atol=tol,
+    )
+
+    m = dpt.logsumexp(x, axis=1)
+    assert_allclose(
+        dpt.asnumpy(m),
+        np.logaddexp.reduce(dpt.asnumpy(x), axis=1, dtype=m.dtype),
+        rtol=tol,
+        atol=tol,
+    )