|
13 | 13 | import torch.nn.functional as F
|
14 | 14 | from torch.testing._internal.common_dtype import floating_types_and, floating_and_complex_types_and
|
15 | 15 | 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, \ |
17 | 17 | download_file, parametrize as parametrize_test, subtest, \
|
18 | 18 | instantiate_parametrized_tests, set_default_dtype
|
19 | 19 | 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):
|
2511 | 2511 | @onlyCUDA
|
2512 | 2512 | @largeTensorTest('40GB')
|
2513 | 2513 | @largeTensorTest('24GB', 'cpu')
|
| 2514 | + # Skipped for ROCm temp - https://ontrack-internal.amd.com/browse/SWDEV-383635 |
| 2515 | + @skipIfRocm |
2514 | 2516 | def test_conv3d_64bit_indexing(self, device):
|
2515 | 2517 | x = torch.rand(1, 32, 512, 512, 256)
|
2516 | 2518 | m = torch.nn.Conv3d(32, 1, kernel_size=1, padding=0, stride=1, bias=False)
|
|
0 commit comments