Skip to content

Commit e90934f

Browse files
ethanwee1jithunnair-amd
authored andcommitted
[rocm7.0_internal_testing] Update test_binary_ufuncs.py after numpy upgrade (#2288)
1 parent a7d3b2e commit e90934f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/test_binary_ufuncs.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1447,7 +1447,7 @@ def to_np(value):
14471447
try:
14481448
np_res = np.power(to_np(base), to_np(np_exponent))
14491449
expected = (
1450-
torch.from_numpy(np_res)
1450+
torch.from_numpy(np_res).to(dtype=base.dtype)
14511451
if isinstance(np_res, np.ndarray)
14521452
else torch.tensor(np_res, dtype=base.dtype)
14531453
)

0 commit comments

Comments
 (0)