Skip to content

Commit d4e4d41

Browse files
praguptapruthvistony
authored andcommitted
Skip test_mm_triton_kernel_benchmark (#1376)
* Running triton kernel on ROCM only has one GB/s metric reported * Update test_kernel_benchmark.py
1 parent 87dda1b commit d4e4d41

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

test/inductor/test_kernel_benchmark.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
from torch._inductor.utils import fresh_inductor_cache
1313
from torch.testing import FileCheck
1414
from torch.testing._internal.inductor_utils import GPU_TYPE, HAS_GPU
15-
15+
from torch.testing._internal.common_utils import skipIfRocm
1616

1717
class TestKernelBenchmark(TestCase):
1818
@classmethod
@@ -97,6 +97,7 @@ def f(a, b):
9797

9898
@config.patch(max_autotune=True, max_autotune_gemm_backends="TRITON")
9999
@fresh_inductor_cache()
100+
@skipIfRocm #This seems to be disabled upstream https://github.com/pytorch/pytorch/issues/118346
100101
def test_mm_triton_kernel_benchmark(self):
101102
M = 2048
102103
N = 2432

0 commit comments

Comments
 (0)