Skip to content

Commit 47477e3

Browse files
committed
Temporarily skip test_conv3d_64bit_indexing
- Rocblas API support is requested - SWDEV-383635 & sub task - SWDEV-390218
1 parent 3f14efa commit 47477e3

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

test/nn/test_convolution.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,7 @@
5555
parametrize as parametrize_test,
5656
run_tests,
5757
set_default_dtype,
58+
skipIfRocm,
5859
skipIfNotMiopenSuggestNHWC,
5960
skipIfRocmVersionLessThan,
6061
subtest,
@@ -3980,8 +3981,10 @@ def test_conv_double_backward_strided_with_3D_input_and_weight(self, device):
39803981
self.assertEqual(grad_weight.shape, weight.shape)
39813982

39823983
@onlyCUDA
3983-
@largeTensorTest("40GB")
3984-
@largeTensorTest("24GB", "cpu")
3984+
@largeTensorTest('40GB')
3985+
@largeTensorTest('24GB', 'cpu')
3986+
# Skipped for ROCm temp - https://ontrack-internal.amd.com/browse/SWDEV-383635
3987+
@skipIfRocm
39853988
def test_conv3d_64bit_indexing(self, device):
39863989
x = torch.rand(1, 32, 512, 512, 256)
39873990
m = torch.nn.Conv3d(32, 1, kernel_size=1, padding=0, stride=1, bias=False)

0 commit comments

Comments
 (0)