correct order when calling back into AD

JuliaDiff · Apr 5, 2024 · 6102c01 · 6102c01
1 parent 9e1b91c
commit 6102c01
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/stage1/forward.jl b/src/stage1/forward.jl
@@ -221,7 +221,7 @@ for V in (false, true)
 end
 
 function ChainRulesCore.frule_via_ad(::DiffractorRuleConfig, partials, args...)
-    bundles = map(bundle, partials, args)
+    bundles = map(bundle, args, partials)
     result = ∂☆internal{1,false}()(bundles...)
     primal(result), first_partial(result)
 end