Index: mlir/lib/IR/AffineMap.cpp
===================================================================
--- mlir/lib/IR/AffineMap.cpp
+++ mlir/lib/IR/AffineMap.cpp
@@ -322,7 +322,7 @@
     exprs.push_back(
         simplifyAffineExpr(e, map.getNumDims(), map.getNumSymbols()));
   }
-  return AffineMap::get(map.getNumDims(), map.getNumSymbols(), exprs, map.getContext());
+  return AffineMap::get(map.getNumDims(), map.getNumSymbols(), exprs);
 }

 AffineMap mlir::removeDuplicateExprs(AffineMap map) {