diff --git a/python/test/unit/language/test_core.py b/python/test/unit/language/test_core.py index 7e6071926..620595ec4 100644 --- a/python/test/unit/language/test_core.py +++ b/python/test/unit/language/test_core.py @@ -1635,6 +1635,8 @@ def test_libdevice_tensor(dtype_str, expr, lib_path): @pytest.mark.parametrize("dtype_str, expr, lib_path", [('float32', 'libdevice.pow', '')]) def test_libdevice_scalar(dtype_str, expr, lib_path): + if torch.version.hip is not None: + pytest.skip(f"test_libdevice_scalar currently has segfaults on ROCM") @triton.jit def kernel(X, Y, BLOCK: tl.constexpr):