diff --git a/enzyme/Enzyme/CallDerivatives.cpp b/enzyme/Enzyme/CallDerivatives.cpp index 153fbd513e17..94ab990e99e9 100644 --- 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;