more info for shadowRV (#1920)

EnzymeAD · Jun 8, 2024 · cd6f629 · cd6f629
1 parent 79e8299
commit cd6f629
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/enzyme/Enzyme/EnzymeLogic.cpp b/enzyme/Enzyme/EnzymeLogic.cpp
@@ -3023,8 +3023,11 @@ const AugmentedReturn &EnzymeLogic::CreateAugmentedPrimal(
         if (auto ggep = dyn_cast<GetElementPtrInst>(gep)) {
           ggep->setIsInBounds(true);
         }
-        if (!(isa<ConstantExpr>(shadowRV) || isa<ConstantData>(shadowRV))) {
-          shadowRV = VMap[shadowRV];
+        if (!(isa<ConstantExpr>(shadowRV) || isa<ConstantData>(shadowRV) ||
+              isa<ConstantAggregate>(shadowRV))) {
+          auto found = VMap.find(shadowRV);
+          assert(found != VMap.end());
+          shadowRV = found->second;
         }
         if (EnzymeFixupReturn)
           shadowRV = unwrap(EnzymeFixupReturn(wrap(&ib), wrap(shadowRV)));