Skip to content

Commit

Permalink
relax tol
Browse files Browse the repository at this point in the history
  • Loading branch information
masahi committed Jan 21, 2022
1 parent e468604 commit 63a5548
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/python/contrib/test_cudnn.py
Original file line number Diff line number Diff line change
Expand Up @@ -374,8 +374,8 @@ def verify_conv2d_backward_filter(data_dtype, conv_dtype, tensor_format=0, tol=1
@tvm.testing.requires_gpu
@requires_cudnn
def test_conv2d_backward_filter():
verify_conv2d_backward_filter("float32", "float32", tensor_format=0, tol=1e-5)
verify_conv2d_backward_filter("float32", "float32", tensor_format=1, tol=1e-5)
verify_conv2d_backward_filter("float32", "float32", tensor_format=0, tol=1e-4)
verify_conv2d_backward_filter("float32", "float32", tensor_format=1, tol=1e-4)


test_kwargs_default_2d = {
Expand Down

0 comments on commit 63a5548

Please sign in to comment.