Skip to content

Added some rules for scaled airy and bessel functions and incomplete gamma #66

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 9 commits into
base: master
Choose a base branch
from
Open

Added some rules for scaled airy and bessel functions and incomplete gamma #66

Show file tree
Hide file tree
Changes from 5 commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
406d579
added rule for gamma(a, x)
amrods Aug 27, 2021
79ceb6f
added a few rules for airy and bessel functions
amrods Aug 27, 2021
af7dcfd
fixed missing $ in def of zeta
amrods Aug 27, 2021
e206f70
Fixed tests for 2-arg gamma
amrods Sep 6, 2021
b2957af
Merge branch 'JuliaDiff:master' into master
amrods Oct 19, 2021
236ad7a
Merge branch 'master' into master
devmotion Dec 11, 2021
a5ee29d
Update test/runtests.jl
devmotion Dec 11, 2021
ce09b11
Update Project.toml
devmotion Dec 11, 2021
813f4e8
Make rules consistent with ChainRules definitions in SpecialFunctions
devmotion Dec 11, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
23 changes: 15 additions & 8 deletions src/rules.jl
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -141,6 +141,10 @@ end
:( SpecialFunctions.airybiprime($x) )
@define_diffrule SpecialFunctions.airybiprime(x) =
:( $x * SpecialFunctions.airybi($x) )
@define_diffrule SpecialFunctions.airyaix(x) =
:( sqrt($x) * SpecialFunctions.airyaix($x) + SpecialFunctions.airyaiprimex($x) )
@define_diffrule SpecialFunctions.airyaiprimex(x) =
:( $x * SpecialFunctions.airyaix($x) + sqrt($x) * SpecialFunctions.airyaiprimex($x) )
@define_diffrule SpecialFunctions.besselj0(x) =
:( -SpecialFunctions.besselj1($x) )
@define_diffrule SpecialFunctions.besselj1(x) =
Expand All @@ -154,8 +158,6 @@ end
#
# eta
# zeta
# airyaix
# airyaiprimex
# airybix
# airybiprimex

Expand All @@ -170,29 +172,34 @@ end
:NaN, :( (SpecialFunctions.bessely($ν - 1, $x) - SpecialFunctions.bessely($ν + 1, $x)) / 2 )
@define_diffrule SpecialFunctions.besselk(ν, x) =
:NaN, :( -(SpecialFunctions.besselk($ν - 1, $x) + SpecialFunctions.besselk($ν + 1, $x)) / 2 )
@define_diffrule SpecialFunctions.besselkx(ν, x) =
:NaN, :( -(SpecialFunctions.besselkx($ν - 1, $x) - 2SpecialFunctions.besselkx($ν, $x) + SpecialFunctions.besselkx($ν + 1, $x)) / 2 )
@define_diffrule SpecialFunctions.hankelh1(ν, x) =
:NaN, :( (SpecialFunctions.hankelh1($ν - 1, $x) - SpecialFunctions.hankelh1($ν + 1, $x)) / 2 )
@define_diffrule SpecialFunctions.hankelh1x(ν, x) =
:NaN, :( (SpecialFunctions.hankelh1x($ν - 1, $x) - 2im*SpecialFunctions.hankelh1x($ν, $x) - SpecialFunctions.hankelh1x($ν + 1, $x)) / 2 )
@define_diffrule SpecialFunctions.hankelh2(ν, x) =
:NaN, :( (SpecialFunctions.hankelh2($ν - 1, $x) - SpecialFunctions.hankelh2($ν + 1, $x)) / 2 )
@define_diffrule SpecialFunctions.hankelh2x(ν, x) =
:NaN, :( (SpecialFunctions.hankelh2x($ν - 1, $x) + 2im*SpecialFunctions.hankelh2x($ν, $x) - SpecialFunctions.hankelh2x($ν + 1, $x)) / 2 )
@define_diffrule SpecialFunctions.gamma(a, x) =
:NaN, :(-exp(-$x) * $x^($a - 1))
@define_diffrule SpecialFunctions.polygamma(m, x) =
:NaN, :( SpecialFunctions.polygamma($m + 1, $x) )
@define_diffrule SpecialFunctions.beta(a, b) =
:( SpecialFunctions.beta($a, $b)*(SpecialFunctions.digamma($a) - SpecialFunctions.digamma($a + $b)) ), :( SpecialFunctions.beta($a, $b)*(SpecialFunctions.digamma($b) - SpecialFunctions.digamma($a + $b)) )
@define_diffrule SpecialFunctions.logbeta(a, b) =
@define_diffrule SpecialFunctions.logbeta(a, b) =
:( SpecialFunctions.digamma($a) - SpecialFunctions.digamma($a + $b) ), :( SpecialFunctions.digamma($b) - SpecialFunctions.digamma($a + $b) )
@define_diffrule SpecialFunctions.zeta(s, z) =
:NaN, :( -$s * SpecialFunctions.zeta(1 + $s, $z) )

# TODO:
#
# zeta
# besseljx
# besselyx
# besselix
# besselkx
# besselh
# besselhx
# hankelh1x
# hankelh2
# hankelh2x

# ternary #
#---------#
Expand Down
7 changes: 7 additions & 0 deletions test/runtests.jl
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,8 +18,15 @@ end

non_numeric_arg_functions = [(:Base, :rem2pi, 2), (:Base, :ifelse, 3)]

recently_defined_functions = []
if !hasmethod(SpecialFunctions.gamma, Tuple{Number, Number})
# 2-arg gamma added after SpecialFunctions v0.8.0
push!(recently_defined_functions, (:SpecialFunctions, :gamma, 2))
end

for (M, f, arity) in DiffRules.diffrules()
(M, f, arity) ∈ non_numeric_arg_functions && continue
(M, f, arity) ∈ recently_defined_functions && continue
if arity == 1
@test DiffRules.hasdiffrule(M, f, 1)
deriv = DiffRules.diffrule(M, f, :goo)
Expand Down