Skip to content

Commit 3183641

Browse files
committed
fix AD tests
1 parent 09c5721 commit 3183641

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/BesselFunctions/besselk.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -244,7 +244,7 @@ function _besselk(v::T, x::T) where T <: Union{Float32, Float64}
244244

245245
else
246246
v_floor, v_int = modf(v)
247-
if x >= 1.5 # determine cutoff as function for differnet types
247+
if x > 1.5 # determine cutoff as function for differnet types
248248
kv, kvp1 = besselkx_levin(v_floor, x, Val(16)), besselkx_levin(v_floor + 1, x, Val(16))
249249
return besselk_up_recurrence(x, kvp1, kv, v_floor + 1, v)[1] * exp(-x)
250250
else

test/besselk_enzyme_test.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,10 +8,10 @@ dbesselkx_dv(v, x) = autodiff(Forward, _v->besselkx_levin(_v, x, Val(30)),
88
dbesselkx_dx(v, x) = autodiff(Forward, _x->besselkx_levin(v, _x, Val(30)),
99
Duplicated, Duplicated(x, 1.0))[2]
1010

11-
dbesselk_ps_dv(v, x) = autodiff(Forward, _v->besselk_power_series(_v, x),
11+
dbesselk_ps_dv(v, x) = autodiff(Forward, _v->besselk(_v, x),
1212
Duplicated, Duplicated(v, 1.0))[2]
1313

14-
dbesselk_ps_dx(v, x) = autodiff(Forward, _x->besselk_power_series(v, _x),
14+
dbesselk_ps_dx(v, x) = autodiff(Forward, _x->besselk(v, _x),
1515
Duplicated, Duplicated(x, 1.0))[2]
1616

1717

0 commit comments

Comments
 (0)