[TRITON-MLIR][BACKEND] Atomic support mask (#889)

Co-authored-by: dongdongl <dongdongl@nvidia.com>
2022-11-19 19:57:19 +08:00
parent dab4855bdf
commit afaf59b0c9
2 changed files with 15 additions and 5 deletions
--- a/python/tests/test_core.py
+++ b/python/tests/test_core.py
@@ -672,7 +672,7 @@ def test_tuples():
 #     np.testing.assert_allclose(z_ref, to_numpy(z_tri), rtol=1e-4)

 def test_tensor_atomic_rmw_add_elementwise(device="cuda"):
-    shape0, shape1 = 16, 16
+    shape0, shape1 = 2, 8
    @triton.jit
    def kernel(Z, X, SHAPE0: tl.constexpr, SHAPE1: tl.constexpr):
        off0 = tl.arange(0, SHAPE0)