diff --git a/ext/FastPowerEnzymeExt.jl b/ext/FastPowerEnzymeExt.jl index c01b21a..bca131b 100644 --- a/ext/FastPowerEnzymeExt.jl +++ b/ext/FastPowerEnzymeExt.jl @@ -5,9 +5,7 @@ import FastPower: fastpower using Enzyme using Enzyme.EnzymeRules: FwdConfig -Enzyme.EnzymeRules.@easy_rule( - FastPower.fastpower(x, y), - ( y * fastpower(x, y - 1), Ω * log(x) ) -) +Enzyme.EnzymeRules.@easy_rule(FastPower.fastpower(x, y), + (y * fastpower(x, y - 1), Ω * log(x))) end