From 648e4cfe89cc4806a158b879a606436bfbb190a2 Mon Sep 17 00:00:00 2001 From: Michael Melesse Date: Wed, 26 Oct 2022 18:22:23 +0000 Subject: [PATCH] skip test_atomic_rmw on rocm --- python/test/unit/language/test_core.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/python/test/unit/language/test_core.py b/python/test/unit/language/test_core.py index b2d450e2f..0e3d696f0 100644 --- a/python/test/unit/language/test_core.py +++ b/python/test/unit/language/test_core.py @@ -609,6 +609,10 @@ def test_tuples(): ] for mode in ['all_neg', 'all_pos', 'min_neg', 'max_pos']])) def test_atomic_rmw(op, dtype_x_str, mode, device='cuda'): + if torch.version.hip is not None: + # if dtype_x_str in ["uint32","int32","float32"]: + pytest.skip(f"test_atomic_rmw[{dtype_x_str}] currently has segfaults on ROCM") + n_programs = 5 # triton kernel