Skip to content

Commit

Permalink
test float16
Browse files Browse the repository at this point in the history
  • Loading branch information
grlee77 committed Dec 7, 2020
1 parent 322906d commit 6608b60
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -123,7 +123,7 @@ def _entropy(self, xp, scp, dtype, shape, use_qk, base, axis, normalize):
res = xp.asarray(res, xp.float16 if is_float16 else float_type)
return res

@testing.for_all_dtypes(no_float16=True, no_complex=True)
@testing.for_all_dtypes(no_complex=True)
@testing.numpy_cupy_allclose(rtol={cupy.float16: 1e-3,
cupy.float32: 1e-6,
'default': 1e-15},
Expand Down

0 comments on commit 6608b60

Please sign in to comment.