Skip to content

Commit 7dcd160

Browse files
committed
update scaling factor
1 parent 1017bf3 commit 7dcd160

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/test_dot.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -383,7 +383,7 @@ def test_tensordot(self, dtype, axes):
383383

384384
result = dpnp.tensordot(ia, ib, axes=axes)
385385
expected = numpy.tensordot(a, b, axes=axes)
386-
assert_dtype_allclose(result, expected, factor=16)
386+
assert_dtype_allclose(result, expected, factor=24)
387387

388388
@pytest.mark.parametrize("dtype", get_complex_dtypes())
389389
@pytest.mark.parametrize("axes", [-3, -2, -1, 0, 1, 2])
@@ -399,7 +399,7 @@ def test_tensordot_complex(self, dtype, axes):
399399

400400
result = dpnp.tensordot(ia, ib, axes=axes)
401401
expected = numpy.tensordot(a, b, axes=axes)
402-
assert_dtype_allclose(result, expected, factor=16)
402+
assert_dtype_allclose(result, expected, factor=24)
403403

404404
@pytest.mark.parametrize("dtype", get_all_dtypes(no_bool=True))
405405
@pytest.mark.parametrize(
@@ -424,7 +424,7 @@ def test_tensordot_axes(self, dtype, axes):
424424

425425
result = dpnp.tensordot(ia, ib, axes=axes)
426426
expected = numpy.tensordot(a, b, axes=axes)
427-
assert_dtype_allclose(result, expected, factor=16)
427+
assert_dtype_allclose(result, expected, factor=24)
428428

429429
@pytest.mark.parametrize("dtype1", get_all_dtypes())
430430
@pytest.mark.parametrize("dtype2", get_all_dtypes())
@@ -440,7 +440,7 @@ def test_tensordot_input_dtype_matrix(self, dtype1, dtype2):
440440

441441
result = dpnp.tensordot(ia, ib)
442442
expected = numpy.tensordot(a, b)
443-
assert_dtype_allclose(result, expected, factor=16)
443+
assert_dtype_allclose(result, expected, factor=24)
444444

445445
def test_tensordot_strided(self):
446446
for dim in [1, 2, 3, 4]:

0 commit comments

Comments
 (0)