From 5ae464337177200053f0d19bdfcf042ba6a796ba Mon Sep 17 00:00:00 2001 From: William Moses Date: Sat, 7 Dec 2024 11:36:27 -0600 Subject: [PATCH] Update EnzymeSpecialFunctionsExt.jl --- ext/EnzymeSpecialFunctionsExt.jl | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ext/EnzymeSpecialFunctionsExt.jl b/ext/EnzymeSpecialFunctionsExt.jl index 09e62e98c4..5c3b59a265 100644 --- 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