fix bug

DeepLink-org · Dec 6, 2023 · 09676c9 · 09676c9
1 parent 4a2ae4f
commit 09676c9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dipu/tests/python/unittests/test_minimum_maximum.py b/dipu/tests/python/unittests/test_minimum_maximum.py
@@ -49,7 +49,7 @@ def test_maximum_scalar(self):
         # scalar on cpu
         r_dipu1 = torch.maximum(a.to(self.dipu), b)
         # scalar on device
-        r_dipu2 = torch.maximum(a.to(self.dipu), b)
+        r_dipu2 = torch.maximum(a.to(self.dipu), b.to(self.dipu))
         r_cpu = torch.maximum(a, b)
         self.assertEqual(r_dipu1.to(self.cpu), r_cpu)
         self.assertEqual(r_dipu2.to(self.cpu), r_cpu)