Fix ordering for vector fallback of apply iterate (#1062)

EnzymeAD · Nov 7, 2023 · 8814bc0 · 8814bc0
1 parent a716c08
commit 8814bc0
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/compiler.jl b/src/compiler.jl
@@ -2667,6 +2667,9 @@ function common_apply_iterate_fwd(offset, B, orig, gutils, normalR, shadowR)
             shadow = LLVM.UndefValue(ST)
             for i in 1:width
                 shadow = insert_value!(B, shadow, cal, i-1)
+                if i == 1
+                    API.moveBefore(cal, shadow, B)
+                end
             end
         end
         unsafe_store!(shadowR, shadow.ref)
@@ -2690,6 +2693,9 @@ function common_apply_iterate_augfwd(offset, B, orig, gutils, normalR, shadowR,
             shadow = LLVM.UndefValue(ST)
             for i in 1:width
                 shadow = insert_value!(B, shadow, cal, i-1)
+                if i == 1
+                    API.moveBefore(cal, shadow, B)
+                end
             end
         end
         unsafe_store!(shadowR, shadow.ref)