Merge pull request #229 from JuliaDiff/oscardssmith-shuffle-down-taylor

make `shuffle_down` of `TaylorBundle` use `TaylorBundle` rather than `ExplicitBundle`
JuliaDiff · Oct 16, 2023 · 7be0a92 · 7be0a92
2 parents 9cae206 + bba2b88
commit 7be0a92
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/stage1/forward.jl b/src/stage1/forward.jl
@@ -26,10 +26,10 @@ end
 
 function shuffle_down(b::TaylorBundle{N, B}) where {N, B}
     Base.@constprop :aggressive function _sdown(i::Int64)
-        ExplicitTangentBundle{1}(b.tangent.coeffs[i], (b.tangent.coeffs[i+1],))
+        TaylorBundle{1}(b.tangent.coeffs[i], (b.tangent.coeffs[i+1],))
     end
     TaylorBundle{N-1}(
-        ExplicitTangentBundle{1}(b.primal, (b.tangent.coeffs[1],)),
+        TaylorBundle{1}(b.primal, (b.tangent.coeffs[1],)),
         ntuple(_sdown, N-1))
 end