diff --git a/test/nn/test_convolution.py b/test/nn/test_convolution.py index 2687ab5d089c..abc3cffe3d4d 100644 --- a/test/nn/test_convolution.py +++ b/test/nn/test_convolution.py @@ -4041,6 +4041,7 @@ class TestConvolutionNNDeviceType(NNTestCase): @onlyCUDA @largeTensorTest("40GB") @largeTensorTest("24GB", "cpu") + @tf32_on_and_off(0.005) def test_conv3d_64bit_indexing(self, device): x = torch.rand(1, 32, 512, 512, 256) m = torch.nn.Conv3d(32, 1, kernel_size=1, padding=0, stride=1, bias=False)