Skip to content

Commit 172f384

Browse files
authored
Update test_convolution.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 3676831 commit 172f384

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/nn/test_convolution.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@
4848
gradgradcheck,
4949
instantiate_parametrized_tests,
5050
MACOS_VERSION,
51-
NAVI_ARCH,
51+
NAVI4_ARCH,
5252
parametrize as parametrize_test,
5353
run_tests,
5454
set_default_dtype,
@@ -3885,7 +3885,7 @@ def test_conv2d_no_grad(self, device, dtype):
38853885

38863886
@onlyCUDA
38873887
@skipCUDAIfNoCudnn
3888-
@skipIfRocmArch(NAVI_ARCH) # not supported by MIOPEN on NAVI
3888+
@skipIfRocmArch(NAVI4_ARCH) # not supported by MIOPEN on NAVI4x
38893889
@dtypes(torch.float, torch.float16)
38903890
@precisionOverride({torch.half: 0.002, torch.float: 1e-4})
38913891
def test_cudnn_convolution_relu(self, device, dtype):

0 commit comments

Comments
 (0)