Skip to content

Commit 8661299

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

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

test/nn/test_convolution.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
import torch.nn.functional as F
1414
from torch.testing._internal.common_dtype import floating_types_and, floating_and_complex_types_and
1515
from torch.testing._internal.common_utils import run_tests, \
16-
skipIfRocmVersionLessThan, skipIfNotMiopenSuggestNHWC, TEST_SCIPY, TEST_WITH_ROCM, \
16+
skipIfRocm, skipIfRocmVersionLessThan, skipIfNotMiopenSuggestNHWC, TEST_SCIPY, TEST_WITH_ROCM, \
1717
download_file, parametrize as parametrize_test, subtest, \
1818
instantiate_parametrized_tests, set_default_dtype
1919
from torch.testing._internal.common_cuda import TEST_CUDA, TEST_CUDNN
@@ -2511,6 +2511,8 @@ def test_conv_double_backward_strided_with_3D_input_and_weight(self, device):
25112511
@onlyCUDA
25122512
@largeTensorTest('40GB')
25132513
@largeTensorTest('24GB', 'cpu')
2514+
# Skipped for ROCm temp - https://ontrack-internal.amd.com/browse/SWDEV-383635
2515+
@skipIfRocm
25142516
def test_conv3d_64bit_indexing(self, device):
25152517
x = torch.rand(1, 32, 512, 512, 256)
25162518
m = torch.nn.Conv3d(32, 1, kernel_size=1, padding=0, stride=1, bias=False)

0 commit comments

Comments
 (0)