Skip to content

Commit 6d422c7

Browse files
authored
Skipping test on branch with no MIOpen immediate mode support (#1130)
Support for MIOpen immediate mode is not enabled in release/1.11 release/1.12 branches which is required for the test_Conv2d_naive_groups test to pass - https://ontrack-internal.amd.com/browse/SWDEV-355273
1 parent 9940e93 commit 6d422c7

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

test/test_nn.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17537,6 +17537,8 @@ def test_multihead_attention_dtype(self, device, dtype):
1753717537
self.assertEqual(q.size(), out[0].size())
1753817538
self.assertEqual(dtype, out[0].dtype)
1753917539

17540+
# Skip the test for ROCm as per https://ontrack-internal.amd.com/browse/SWDEV-355273
17541+
@skipIfRocm
1754017542
@dtypesIfCUDA(*get_all_fp_dtypes(include_bfloat16=AMPERE_OR_ROCM))
1754117543
@dtypes(torch.float)
1754217544
def test_Conv2d_naive_groups(self, device, dtype):

0 commit comments

Comments
 (0)