1 files changed, 3 insertions, 3 deletions
diff --git a/mlir/lib/Transforms/SimplifyAffineExpr.cpp b/mlir/lib/Transforms/SimplifyAffineExpr.cpp
index 30d3908ac33..a3b8ebf1ba8 100644
--- a/mlir/lib/Transforms/SimplifyAffineExpr.cpp
+++ b/mlir/lib/Transforms/SimplifyAffineExpr.cpp
@@ -59,10 +59,10 @@ PassResult SimplifyAffineExpr::runOnMLFunction(MLFunction *f) {
     void visitOperationStmt(OperationStmt *opStmt) {
       for (auto attr : opStmt->getAttrs()) {
         if (auto *mapAttr = dyn_cast<AffineMapAttr>(attr.second)) {
-          MutableAffineMap mMap(mapAttr->getValue(), context);
+          MutableAffineMap mMap(mapAttr->getValue());
           mMap.simplify();
-          auto *map = mMap.getAffineMap();
-          opStmt->setAttr(attr.first, AffineMapAttr::get(map, context));
+          auto map = mMap.getAffineMap();
+          opStmt->setAttr(attr.first, AffineMapAttr::get(map));
         }
       }
     }