use no_float16=True
@@ -51,11 +51,10 @@ 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()
+ @testing.for_all_dtypes(no_float16=True)
@testing.numpy_cupy_allclose(rtol=1e-3, atol=1e-4, type_check=False)
def test_norm(self, xp, dtype):
a = testing.shaped_arange(self.shape, xp, dtype)
- with testing.NumpyError(divide='ignore'):
return xp.linalg.norm(a, self.ord, self.axis, self.keepdims)
