Update test/nn/test_convolution.py per reviewer's comment

Xia-Weiwen · malfet · web-flow · commit 0873e19fc839 · 2024-03-06T17:10:46.000-08:00
Co-authored-by: Nikita Shulga &lt;2453524+malfet@users.noreply.github.com&gt;
diff --git a/test/nn/test_convolution.py b/test/nn/test_convolution.py
@@ -845,7 +845,7 @@ def _test_conv2d(stride, kernel_size, groups, dilation):
 
     def test_permute_conv2d_issue_120211(self):
         def reproducer(radius: int):
-            image = torch.rand([1, 1024, 1024, 3], dtype=torch.float32, device=torch.device('cpu'))
+            image = torch.rand(1, 1024, 1024, 3)
             image = image.permute(0, 3, 1, 2)
             kernel_x = torch.zeros([3, 1, 1, radius * 2 + 1], device=image.device)
             image = torch.nn.functional.conv2d(image, kernel_x, groups=image.shape[-3])
