lint

eqy · pytorchmergebot · commit 58267d4361a2 · 2025-07-03T04:02:14.000Z
diff --git a/test/nn/test_convolution.py b/test/nn/test_convolution.py
@@ -4057,8 +4057,12 @@ def test_conv3d_64bit_indexing(self, device):
     @largeTensorTest("20GB")
     @largeTensorTest("80GB", "cpu")
     def test_depthwise_conv_64bit_indexing(self, device):
-        x = torch.randn(1, 2, 32800, 32800, dtype=torch.bfloat16).to(memory_format=torch.channels_last)
-        c = nn.Conv2d(2, 2, kernel_size=3, stride=1, padding=1, groups=2, dtype=torch.bfloat16).to(memory_format=torch.channels_last)
+        x = torch.randn(1, 2, 32800, 32800, dtype=torch.bfloat16).to(
+            memory_format=torch.channels_last
+        )
+        c = nn.Conv2d(
+            2, 2, kernel_size=3, stride=1, padding=1, groups=2, dtype=torch.bfloat16
+        ).to(memory_format=torch.channels_last)
         yref = c(x)
         y = c.to(device=device)(x.to(device=device))
         self.assertEqual(yref, y, atol=1e-3, rtol=1e-4)
