fix

EnzymeAD · Oct 5, 2024 · 2be8688 · 2be8688
1 parent 65bec31
commit 2be8688
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/enzyme/Enzyme/MLIR/Implementations/CoreDialectsAutoDiffImplementations.cpp b/enzyme/Enzyme/MLIR/Implementations/CoreDialectsAutoDiffImplementations.cpp
@@ -409,7 +409,7 @@ LogicalResult mlir::enzyme::detail::controlFlowForwardHandler(
   // Replace all uses of original results
   gutils->replaceOrigOpWith(op, reps);
   gutils->erase(newOp);
-  gutils->originalToNewFnOps.map(op, replacement);
+  gutils->originalToNewFnOps[op] = replacement;
 
   return success();
 }