fix(test): add .cpu() before .numpy() for GPU-compatible activation tests

Han Wang · Han Wang · commit 479900d075c5 · 2026-04-18T13:25:13.000+08:00
diff --git a/source/tests/consistent/test_activation.py b/source/tests/consistent/test_activation.py
@@ -113,7 +113,7 @@ def test_pt_expt_consistent_with_ref(self) -> None:
             x = torch.tensor(
                 self.random_input, dtype=torch.float64, device=PT_EXPT_DEVICE
             )
-            test = _torch_activation(x, self.activation).detach().numpy()
+            test = _torch_activation(x, self.activation).detach().cpu().numpy()
             np.testing.assert_allclose(self.ref, test, atol=1e-10)
 
 
@@ -155,5 +155,5 @@ def test_pt_expt_consistent_with_ref(self) -> None:
             x = torch.tensor(
                 self.random_input, dtype=torch.float64, device=PT_EXPT_DEVICE
             )
-            test = _torch_activation(x, self.activation).detach().numpy()
+            test = _torch_activation(x, self.activation).detach().cpu().numpy()
             np.testing.assert_allclose(self.ref, test, atol=1e-10)

Original file line number	Diff line number	Diff line change
`@@ -113,7 +113,7 @@ def test_pt_expt_consistent_with_ref(self) -> None:`
`113`	`113`	`x = torch.tensor(`
`114`	`114`	`self.random_input, dtype=torch.float64, device=PT_EXPT_DEVICE`
`115`	`115`	`)`
`116`		`- test = _torch_activation(x, self.activation).detach().numpy()`
	`116`	`+ test = _torch_activation(x, self.activation).detach().cpu().numpy()`
`117`	`117`	`np.testing.assert_allclose(self.ref, test, atol=1e-10)`
`118`	`118`
`119`	`119`
`@@ -155,5 +155,5 @@ def test_pt_expt_consistent_with_ref(self) -> None:`
`155`	`155`	`x = torch.tensor(`
`156`	`156`	`self.random_input, dtype=torch.float64, device=PT_EXPT_DEVICE`
`157`	`157`	`)`
`158`		`- test = _torch_activation(x, self.activation).detach().numpy()`
	`158`	`+ test = _torch_activation(x, self.activation).detach().cpu().numpy()`
`159`	`159`	`np.testing.assert_allclose(self.ref, test, atol=1e-10)`