Skip to content

Commit df5f4cf

Browse files
committed
multihead_attention: set slightly higher margin for fp imprecision (due to IEEE_DENORMAL)
1 parent 4c92e9c commit df5f4cf

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/test_multihead_attention_layer.f90

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ function allclose(x, y) result(res)
4343
real, intent(in) :: y(:)
4444
logical :: res
4545

46-
res = all(abs(x - y) <= (1e-08 + 1e-05 * abs(y)))
46+
res = all(abs(x - y) <= (1e-06 + 1e-05 * abs(y)))
4747
end function allclose
4848

4949
subroutine set_weights(attention)

0 commit comments

Comments
 (0)