Skip to content

Commit 3676831

Browse files
authored
Update test_freezing.py
For both tests (nn and jit) we need support of `ConvBinWinogradRxSf2x3g1Fused` kernel in MIOpen which currently doesn't support for NAVI4(gfx12x), but it works for NAVI3(gfx11x), so enable tests for NAVI3 but keep disabled for NAVI4.
1 parent 3f3b83a commit 3676831

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/jit/test_freezing.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
from torch.testing._internal.common_quantized import override_quantized_engine
1717
from torch.testing._internal.common_utils import (
1818
raise_on_run_directly,
19-
NAVI_ARCH,
19+
NAVI4_ARCH,
2020
set_default_dtype,
2121
skipCUDAMemoryLeakCheckIf,
2222
skipIfRocmArch,
@@ -2969,7 +2969,7 @@ def test_conv_to_mkldnn_no_mkldnn(self):
29692969
self.assertEqual(frozen(inp), mod(inp))
29702970

29712971
@unittest.skipIf(not (TEST_CUDNN or TEST_WITH_ROCM), "requires CUDNN")
2972-
@skipIfRocmArch(NAVI_ARCH) # not supported by MIOPEN on NAVI
2972+
@skipIfRocmArch(NAVI4_ARCH) # not supported by MIOPEN on NAVI4x
29732973
def test_freeze_conv_relu_fusion(self):
29742974
with set_default_dtype(torch.float):
29752975
conv_bias = [True, False]

0 commit comments

Comments
 (0)