diff --git a/test/test_torch_compile.py b/test/test_torch_compile.py index 3c607bf2e..d6af5d31a 100644 --- a/test/test_torch_compile.py +++ b/test/test_torch_compile.py @@ -10,12 +10,14 @@ from helion._testing import TestCase from helion._testing import skipIfPy314 from helion._testing import skipIfRefEager +from helion._testing import skipIfRocm import helion.language as hl class TestTorchCompile(RefEagerTestBase, TestCase): @skipIfRefEager("does not work with ref eager") @skipIfPy314("torch.compile not yet supported on Python 3.14") + @skipIfRocm("torch.compile add kernel missing kernel metadata fields on ROCm") def test_add_kernel(self): @helion.kernel(config=helion.Config(block_sizes=[1, 2])) def add(x: torch.Tensor, y: torch.Tensor) -> torch.Tensor: