Update EnzymeSpecialFunctionsExt.jl

EnzymeAD · Dec 7, 2024 · 5ae4643 · 5ae4643
1 parent 8e10a0a
commit 5ae4643
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ext/EnzymeSpecialFunctionsExt.jl b/ext/EnzymeSpecialFunctionsExt.jl
@@ -5,7 +5,8 @@ using Enzyme
 
 function __init__()
     Enzyme.Compiler.known_ops[typeof(SpecialFunctions._logabsgamma)] = (:logabsgamma, 1, (:digamma, typeof(SpecialFunctions.digamma)))
-    Enzyme.Compiler.cmplx_known_ops[typeof(SpecialFunctions.bessely)] = (:cmplx_jn, 2, nothing)
+    Enzyme.Compiler.cmplx_known_ops[typeof(SpecialFunctions.bessely)] = (:cmplx_yn, 2, nothing)
+    Enzyme.Compiler.cmplx_known_ops[typeof(SpecialFunctions.besseli)] = (:cmplx_jn, 2, nothing)
     Enzyme.Compiler.cmplx_known_ops[typeof(SpecialFunctions.besselj)] = (:cmplx_jn, 2, nothing)
     Enzyme.Compiler.cmplx_known_ops[typeof(SpecialFunctions.besselk)] = (:cmplx_kn, 2, nothing)
 end