We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 172f384 commit cd1d751Copy full SHA for cd1d751
test/jit/test_freezing.py
@@ -3032,7 +3032,7 @@ def forward(self, x):
3032
self.assertEqual(mod_eager(inp), frozen_mod(inp))
3033
3034
@unittest.skipIf(not (TEST_CUDNN or TEST_WITH_ROCM), "requires CUDNN")
3035
- @skipIfRocmArch(NAVI_ARCH) # not supported by MIOPEN on NAVI
+ @skipIfRocmArch(NAVI4_ARCH) # not supported by MIOPEN on NAVI4x
3036
def test_freeze_conv_relu_fusion_not_forward(self):
3037
with set_default_dtype(torch.float):
3038
0 commit comments