Merge pull request #260 from JuliaDiff/ox/not_explict_order_1

Don't use ExplictTangentBundle in frule_via_ad
JuliaDiff · Feb 9, 2024 · 2621762 · 2621762
2 parents 0b5a05a + e722d6b
commit 2621762
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/stage1/forward.jl b/src/stage1/forward.jl
@@ -126,7 +126,7 @@ function _frule(::NTuple{<:Any, AbstractZero}, f, primal_args...)
 end
 
 function ChainRulesCore.frule_via_ad(::DiffractorRuleConfig, partials, args...)
-    bundles = map((p,a) -> ExplicitTangentBundle{1}(a, (p,)), partials, args)
+    bundles = map(bundle, partials, args)
     result = ∂☆internal{1}()(bundles...)
     primal(result), first_partial(result)
 end