Skip to content

Commit cd1d751

Browse files
authored
Update test_freezing.py
Enable for Navi3x
1 parent 172f384 commit cd1d751

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/jit/test_freezing.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3032,7 +3032,7 @@ def forward(self, x):
30323032
self.assertEqual(mod_eager(inp), frozen_mod(inp))
30333033

30343034
@unittest.skipIf(not (TEST_CUDNN or TEST_WITH_ROCM), "requires CUDNN")
3035-
@skipIfRocmArch(NAVI_ARCH) # not supported by MIOPEN on NAVI
3035+
@skipIfRocmArch(NAVI4_ARCH) # not supported by MIOPEN on NAVI4x
30363036
def test_freeze_conv_relu_fusion_not_forward(self):
30373037
with set_default_dtype(torch.float):
30383038

0 commit comments

Comments
 (0)