@@ -383,7 +383,7 @@ def test_tensordot(self, dtype, axes):
383
383
384
384
result = dpnp .tensordot (ia , ib , axes = axes )
385
385
expected = numpy .tensordot (a , b , axes = axes )
386
- assert_dtype_allclose (result , expected , factor = 16 )
386
+ assert_dtype_allclose (result , expected , factor = 24 )
387
387
388
388
@pytest .mark .parametrize ("dtype" , get_complex_dtypes ())
389
389
@pytest .mark .parametrize ("axes" , [- 3 , - 2 , - 1 , 0 , 1 , 2 ])
@@ -399,7 +399,7 @@ def test_tensordot_complex(self, dtype, axes):
399
399
400
400
result = dpnp .tensordot (ia , ib , axes = axes )
401
401
expected = numpy .tensordot (a , b , axes = axes )
402
- assert_dtype_allclose (result , expected , factor = 16 )
402
+ assert_dtype_allclose (result , expected , factor = 24 )
403
403
404
404
@pytest .mark .parametrize ("dtype" , get_all_dtypes (no_bool = True ))
405
405
@pytest .mark .parametrize (
@@ -424,7 +424,7 @@ def test_tensordot_axes(self, dtype, axes):
424
424
425
425
result = dpnp .tensordot (ia , ib , axes = axes )
426
426
expected = numpy .tensordot (a , b , axes = axes )
427
- assert_dtype_allclose (result , expected , factor = 16 )
427
+ assert_dtype_allclose (result , expected , factor = 24 )
428
428
429
429
@pytest .mark .parametrize ("dtype1" , get_all_dtypes ())
430
430
@pytest .mark .parametrize ("dtype2" , get_all_dtypes ())
@@ -440,7 +440,7 @@ def test_tensordot_input_dtype_matrix(self, dtype1, dtype2):
440
440
441
441
result = dpnp .tensordot (ia , ib )
442
442
expected = numpy .tensordot (a , b )
443
- assert_dtype_allclose (result , expected , factor = 16 )
443
+ assert_dtype_allclose (result , expected , factor = 24 )
444
444
445
445
def test_tensordot_strided (self ):
446
446
for dim in [1 , 2 , 3 , 4 ]:
0 commit comments