remove .cpu() call

NicolasHug · NicolasHug · commit 89f1547ab596 · 2025-10-25T01:36:37.000+01:00
diff --git a/test/test_decoders.py b/test/test_decoders.py
@@ -1725,7 +1725,7 @@ def test_beta_cuda_interface_cpu_fallback(self):
         )
         beta_frame = beta_dec.get_frame_at(0)
 
-        assert psnr(ref_frames.data.cpu(), beta_frame.data.cpu()) > 25
+        assert psnr(ref_frames.data, beta_frame.data) > 25
 
     @needs_cuda
     def test_beta_cuda_interface_error(self):

Original file line number	Diff line number	Diff line change
`@@ -1725,7 +1725,7 @@ def test_beta_cuda_interface_cpu_fallback(self):`
`1725`	`1725`	`)`
`1726`	`1726`	`beta_frame = beta_dec.get_frame_at(0)`
`1727`	`1727`
`1728`		`- assert psnr(ref_frames.data.cpu(), beta_frame.data.cpu()) > 25`
	`1728`	`+ assert psnr(ref_frames.data, beta_frame.data) > 25`
`1729`	`1729`
`1730`	`1730`	`@needs_cuda`
`1731`	`1731`	`def test_beta_cuda_interface_error(self):`