Skip to content

Commit 265c812

Browse files
authored
Skipping test on branch with no MIOpen immediate mode support (pytorch#1175)
Support for MIOpen immediate mode is not enabled in release branches required for the test_Conv2d_naive_groups test to pass https://ontrack-internal.amd.com/browse/SWDEV-355273 cherry-pick from ROCm@ad6db00
1 parent 31ed9c9 commit 265c812

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
@@ -16266,6 +16266,8 @@ def test_multihead_attention_dtype(self, device, dtype):
1626616266
self.assertEqual(q.size(), out[0].size())
1626716267
self.assertEqual(dtype, out[0].dtype)
1626816268

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

0 commit comments

Comments
 (0)