Don't eval in __init__

EnzymeAD · May 27, 2024 · 4ccff4b · 4ccff4b
1 parent c85e63e
commit 4ccff4b
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/ext/ReactantNNlibExt.jl b/ext/ReactantNNlibExt.jl
@@ -3,8 +3,7 @@ module ReactantNNlibExt
 using NNlib
 using Reactant
 
-function __init__()
-for (jlop, hloop) in ((:(NNlib.tanh), :tanh),(:(NNlib.tanh_fast), :tanh),)
+for (jlop, hloop) in ((:(NNlib.tanh), :tanh), (:(NNlib.tanh_fast), :tanh),)
     @eval begin
         if $jlop != Base.tanh && $jlop != Base.FastMath.tanh_fast
             function Reactant.elem_apply(::typeof($jlop), lhs::Reactant.TracedRArray{ElType,Shape,N}) where {ElType,Shape,N}
@@ -13,10 +12,9 @@ for (jlop, hloop) in ((:(NNlib.tanh), :tanh),(:(NNlib.tanh_fast), :tanh),)
         end
     end
 end
-end
 
 # TODO handle non finite cases
-function NNlib.softmax!(out::Reactant.TracedRArray{T, Shape, N}, x::AbstractArray; dims = 1) where {T, Shape, N}
+function NNlib.softmax!(out::Reactant.TracedRArray{T,Shape,N}, x::AbstractArray; dims=1) where {T,Shape,N}
     max_ = NNlib.fast_maximum(x; dims)
     #if all(isfinite, max_)
         @fastmath out .= exp.(x .- max_)