rename _lookup_grad

FluxML · Jun 18, 2021 · 8d9fac7 · 8d9fac7
1 parent d949838
commit 8d9fac7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/compiler/emit.jl b/src/compiler/emit.jl
@@ -95,7 +95,7 @@ end
 
 varargs(m::Method, n) = m.isva ? n - m.nargs + 1 : nothing
 
-function _lookup_grad(T)
+function _generate_pullback_via_decomposition(T)
   (m = meta(T)) === nothing && return
   va = varargs(m.method, length(T.parameters))
   forw, back = stacks!(Adjoint(IR(m), varargs = va, normalise = false), T)

diff --git a/src/compiler/interface2.jl b/src/compiler/interface2.jl
@@ -23,7 +23,7 @@ end
 
   hascr && return :($chain_rrule_f(ZygoteRuleConfig(ctx), f, args...))
 
-  g = try _lookup_grad(T) catch e e end
+  g = try _generate_pullback_via_decomposition(T) catch e e end
   g === nothing && return :(f(args...), Pullback{$T}((f,)))
   meta, forw, _ = g
   argnames!(meta, Symbol("#self#"), :ctx, :f, :args)
@@ -37,7 +37,8 @@ end
 
 @generated function (j::Pullback{T})(Δ) where T
   ignore_sig(T) && return :nothing
-  g = try _lookup_grad(T)
+  g = try 
+    _generate_pullback_via_decomposition(T)
   catch e
     rethrow(CompileError(T,e))
   end