@@ -186,13 +186,13 @@ function scalar_rrule_expr(__source__, f, call, setup_stmts, inputs, partials)
186186 Δs = _propagator_inputs (n_outputs)
187187
188188 # Make a projector for each argument
189- ps, setup_ps = _make_projectors (call. args[2 : end ])
190- append! (setup_stmts, setup_ps )
189+ projs, psetup = _make_projectors (call. args[2 : end ])
190+ append! (setup_stmts, psetup )
191191
192192 # 1 partial derivative per input
193193 pullback_returns = map (1 : n_inputs) do input_i
194194 ∂s = [partial. args[input_i] for partial in partials]
195- propagation_expr (Δs, ∂s, true , ps [input_i])
195+ propagation_expr (Δs, ∂s, true , projs [input_i])
196196 end
197197
198198 # Multi-output functions have pullbacks with a tuple input that will be destructured
@@ -221,9 +221,9 @@ _propagator_inputs(n) = [esc(gensym(Symbol(:Δ, i))) for i in 1:n]
221221
222222" given the variable names, escaped but without types, makes setup expressions for projection operators"
223223function _make_projectors (xs)
224- ps = map (x -> gensym (Symbol (:proj_ , x. args[1 ])), xs)
225- setups = map ((x,p) -> :($ p = ProjectTo ($ x)), xs, ps )
226- return ps , setups
224+ projs = map (x -> gensym (Symbol (:proj_ , x. args[1 ])), xs)
225+ setups = map ((x,p) -> :($ p = ProjectTo ($ x)), xs, projs )
226+ return projs , setups
227227end
228228
229229"""
0 commit comments