fix decorator location

eqy · web-flow · commit 18ba66e786ca · 2025-05-15T07:54:38.000-07:00
diff --git a/test/nn/test_convolution.py b/test/nn/test_convolution.py
@@ -4042,7 +4042,6 @@ def test_conv_double_backward_strided_with_3D_input_and_weight(self, device):
         self.assertEqual(grad_input.shape, input.shape)
         self.assertEqual(grad_weight.shape, weight.shape)
 
-    @skipCUDAIfRocm
     @onlyCUDA
     @largeTensorTest("40GB")
     @largeTensorTest("24GB", "cpu")
@@ -4053,6 +4052,7 @@ def test_conv3d_64bit_indexing(self, device):
         y = m.to(device=device)(x.to(device=device))
         self.assertEqual(yref, y)
 
+    @skipCUDAIfRocm
     @onlyCUDA
     @largeTensorTest("20GB")
     @largeTensorTest("80GB", "cpu")
