Test output dtype of `linalg.norm`
This reverts commit
NumPy 1.14 has been dropped.
@@ -49,11 +49,8 @@ class TestTrace(unittest.TestCase):
@testing.gpu
class TestNorm(unittest.TestCase):
- # TODO(kmaehashi) Currently dtypes returned from CuPy is not compatible
- # with NumPy. We should remove `type_check=False` once NumPy is fixed.
- # See https://github.com/cupy/cupy/pull/875 for details.
@testing.for_all_dtypes(no_float16=True)
- @testing.numpy_cupy_allclose(rtol=1e-3, atol=1e-4, type_check=False)
+ @testing.numpy_cupy_allclose(rtol=1e-3, atol=1e-4)
def test_norm(self, xp, dtype):
a = testing.shaped_arange(self.shape, xp, dtype)
return xp.linalg.norm(a, self.ord, self.axis, self.keepdims)
