Merge pull request #40284 from ROCmSoftwarePlatform:google_upstream_rocm_platform_fix_200608
PiperOrigin-RevId: 315492021 Change-Id: I3ff1493149b8a16786bb3a342198c50768e3795f
This commit is contained in:
commit
a866c95a1a
@ -159,7 +159,7 @@ class MathTest(test.TestCase, parameterized.TestCase):
|
||||
actual.shape, expected.shape,
|
||||
'Shape mismatch.\nActual: {}\nExpected: {}\n{}'.format(
|
||||
actual.shape, expected.shape, msg))
|
||||
np.testing.assert_almost_equal(actual.tolist(), expected.tolist())
|
||||
np.testing.assert_allclose(actual.tolist(), expected.tolist(), rtol=1e-6)
|
||||
|
||||
def testArgsort(self):
|
||||
self._testUnaryOp(np_math_ops.argsort, np.argsort, 'argsort')
|
||||
|
Loading…
Reference in New Issue
Block a user