Skip to content

Commit 83364f0

Browse files
committed
Skipping test_complex_half_reference_testing for conv 3D operator
1 parent 4388e4a commit 83364f0

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

test/test_ops.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@
3131
set_default_dtype,
3232
suppress_warnings,
3333
noncontiguous_like,
34+
TEST_WITH_ROCM,
3435
TEST_WITH_ASAN,
3536
TEST_WITH_UBSAN,
3637
IS_WINDOWS,
@@ -1177,6 +1178,10 @@ def _test_inplace_preserve_storage(samples, variants):
11771178
# NOTE: We test against complex64 as NumPy doesn't have a complex32 equivalent dtype.
11781179
@ops(op_db, allowed_dtypes=(torch.complex32,))
11791180
def test_complex_half_reference_testing(self, device, dtype, op):
1181+
# SWDEV-373709, refer for more info
1182+
if all([TEST_WITH_ROCM, op.name == "nn.functional.conv_transpose3d", device == 'cuda:0', dtype == torch.complex32]):
1183+
self.skipTest("Skipped!!! SWDEV-373709 Tensor accuracy issue")
1184+
11801185
if not op.supports_dtype(torch.complex32, device):
11811186
unittest.skip("Does not support complex32")
11821187

0 commit comments

Comments
 (0)