File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -21,7 +21,7 @@ end function
21
21
elemental ${rt}$ module function gaussian_grad_${rk}$( x ) result( y )
22
22
${rt}$, intent(in) :: x
23
23
24
- y = -2_ ${rk}$ * x * exp(-x**2)
24
+ y = -2._ ${rk}$ * x * exp(-x**2)
25
25
end function
26
26
27
27
#:endfor
@@ -99,13 +99,13 @@ end function
99
99
elemental ${rt}$ module function gelu_approx_${rk}$( x ) result( y )
100
100
${rt}$, intent(in) :: x
101
101
102
- y = 0.5_${rk}$ * x * (1 + ferf(x * isqrt2_${rk}$))
102
+ y = 0.5_${rk}$ * x * (1._${rk}$ + ferf(x * isqrt2_${rk}$))
103
103
end function
104
104
105
105
elemental ${rt}$ module function gelu_approx_grad_${rk}$( x ) result( y )
106
106
${rt}$, intent(in) :: x
107
107
108
- y = 0.5_${rk}$ * (1 + ferf(x * isqrt2_${rk}$) )
108
+ y = 0.5_${rk}$ * (1._${rk}$ + ferf(x * isqrt2_${rk}$) )
109
109
y = y + x * isqrt2_${rk}$ * exp( - 0.5_${rk}$ * x**2 )
110
110
end function
111
111
@@ -334,7 +334,7 @@ elemental ${rt}$ module function ferf_${rk}$( x ) result( y )
334
334
${rt}$ :: abs_x
335
335
336
336
abs_x = abs(x)
337
- y = 1._${rk}$ - 1._${rk}$ / (1+ 0.278393_${rk}$*abs_x + 0.230389_${rk}$*abs_x**2 + 0.000972_${rk}$*abs_x**3 + 0.078108_${rk}$*abs_x**4)**4
337
+ y = 1._${rk}$ - 1._${rk}$ / (1._${rk}$ + 0.278393_${rk}$*abs_x + 0.230389_${rk}$*abs_x**2 + 0.000972_${rk}$*abs_x**3 + 0.078108_${rk}$*abs_x**4)**4
338
338
y = y * sign(1.0_${rk}$,x)
339
339
end function
340
340
You can’t perform that action at this time.
0 commit comments