Fix zero on non pointer allocation (#1938)

EnzymeAD · Jun 23, 2024 · e422ba2 · e422ba2
1 parent 7fee772
commit e422ba2
Showing 1 changed file with 12 additions and 12 deletions.
diff --git a/enzyme/Enzyme/CallDerivatives.cpp b/enzyme/Enzyme/CallDerivatives.cpp
@@ -3144,18 +3144,18 @@ bool AdjointGenerator::handleKnownCallDerivatives(
                   if (EnzymeShadowAllocRewrite)
                     EnzymeShadowAllocRewrite(wrap(anti), gutils);
                 }
-                if (Mode == DerivativeMode::ReverseModeCombined ||
-                    (Mode == DerivativeMode::ReverseModePrimal &&
-                     forwardsShadow) ||
-                    (Mode == DerivativeMode::ReverseModeGradient &&
-                     backwardsShadow) ||
-                    (Mode == DerivativeMode::ForwardModeSplit &&
-                     backwardsShadow)) {
-                  if (!inLoop) {
-                    zeroKnownAllocation(bb, anti, args, funcName, gutils->TLI,
-                                        &call);
-                    zeroed = true;
-                  }
+              }
+              if (Mode == DerivativeMode::ReverseModeCombined ||
+                  (Mode == DerivativeMode::ReverseModePrimal &&
+                   forwardsShadow) ||
+                  (Mode == DerivativeMode::ReverseModeGradient &&
+                   backwardsShadow) ||
+                  (Mode == DerivativeMode::ForwardModeSplit &&
+                   backwardsShadow)) {
+                if (!inLoop) {
+                  zeroKnownAllocation(bb, anti, args, funcName, gutils->TLI,
+                                      &call);
+                  zeroed = true;
                 }
               }
               return anti;