Fix random calculation error in test_sum_on_device1 (#994)

Fix random calculation error in test_sum_on_device1 by adding device switch logic.
intel · Oct 22, 2024 · b4a760c · b4a760c
1 parent 88b029e
commit b4a760c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/regressions/test_operation_on_device_1.py b/test/regressions/test_operation_on_device_1.py
@@ -5,6 +5,7 @@ class TestOperationOnDevice1(TestCase):
     def test_sum_on_device1(self, dtype=torch.float):
         if torch.xpu.device_count() >= 2:
             a = torch.randn(2, 3, device=torch.device('xpu:1'))
+            torch.xpu.set_device(1)
             res = a.sum()
             ref = a.cpu().sum()
             self.assertEqual(ref, res)