Fix issue with test/nn/test_convolution:TestConvolutionNNDeviceTypeCUDA.test_conv_large_batch_1_cuda (pytorch#145067)

rec · nWEIdia · pytorchmergebot · commit eaef61368850 · 2025-01-17T20:31:25.000Z
Pull Request resolved: pytorch#145067 Approved by: https://github.com/Skylion007, https://github.com/nWEIdia Co-authored-by: Wei Wang <143543872+nWEIdia@users.noreply.github.com>
diff --git a/test/nn/test_convolution.py b/test/nn/test_convolution.py
@@ -3246,6 +3246,7 @@ def test_conv_large_batch_1(self, device):
             .half()
         )
         output = model(input_tensor)
+        _model_cpu = model.cpu().float()
         output_cpu = model(input_tensor.float().cpu())
         self.assertEqual(output.cpu().float(), output_cpu, atol=1e-3, rtol=1e-3)
 

Original file line number	Diff line number	Diff line change
`@@ -3246,6 +3246,7 @@ def test_conv_large_batch_1(self, device):`
`3246`	`3246`	`.half()`
`3247`	`3247`	`)`
`3248`	`3248`	`output = model(input_tensor)`
	`3249`	`+ _model_cpu = model.cpu().float()`
`3249`	`3250`	`output_cpu = model(input_tensor.float().cpu())`
`3250`	`3251`	`self.assertEqual(output.cpu().float(), output_cpu, atol=1e-3, rtol=1e-3)`
`3251`	`3252`