diff --git a/test/test_binary_ufuncs.py b/test/test_binary_ufuncs.py index ee9fb490356fb0..689f718eb9b683 100644 --- a/test/test_binary_ufuncs.py +++ b/test/test_binary_ufuncs.py @@ -1446,7 +1446,7 @@ def to_np(value): try: np_res = np.power(to_np(base), to_np(np_exponent)) expected = ( - torch.from_numpy(np_res) + torch.from_numpy(np_res).to(dtype=base.dtype) if isinstance(np_res, np.ndarray) else torch.tensor(np_res, dtype=base.dtype) )