|
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
|
@@ -2480,6 +2480,8 @@ def test_conv_double_backward_strided_with_3D_input_and_weight(self, device):
|
2480 | 2480 | @onlyCUDA
|
2481 | 2481 | @largeTensorTest('40GB')
|
2482 | 2482 | @largeTensorTest('24GB', 'cpu')
|
| 2483 | + # Skipped for ROCm temp - https://ontrack-internal.amd.com/browse/SWDEV-383635 |
| 2484 | + @skipIfRocm |
2483 | 2485 | def test_conv3d_64bit_indexing(self, device):
|
2484 | 2486 | x = torch.rand(1, 32, 512, 512, 256)
|
2485 | 2487 | m = torch.nn.Conv3d(32, 1, kernel_size=1, padding=0, stride=1, bias=False)
|
|
0 commit comments