Skip to content

Commit 3ce9b59

Browse files
Use result_type with tensors to take device capability into account
1 parent 15fa952 commit 3ce9b59

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

dpctl/tests/test_usm_ndarray_linalg.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -656,7 +656,7 @@ def test_tensordot_promotion():
656656

657657
t3 = dpt.zeros((10, 10), dtype="u4")
658658
r3 = dpt.tensordot(t1, t3)
659-
assert r3.dtype == dpt.result_type(t1.dtype, t3.dtype)
659+
assert r3.dtype == dpt.result_type(t1, t3)
660660

661661

662662
def test_tensordot_axes_errors():

0 commit comments

Comments
 (0)