address comments and 2024->2025

Signed-off-by: hanhanW <[email protected]>
iree-org · Jan 6, 2025 · a479a4a · a479a4a
1 parent cb19fce
commit a479a4a
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/compiler/src/iree/compiler/Dialect/Stream/Transforms/SpecializeEncodings.cpp b/compiler/src/iree/compiler/Dialect/Stream/Transforms/SpecializeEncodings.cpp
@@ -1,4 +1,4 @@
-// Copyright 2024 The IREE Authors
+// Copyright 2025 The IREE Authors
 //
 // Licensed under the Apache License v2.0 with LLVM Exceptions.
 // See https://llvm.org/LICENSE.txt for license information.
@@ -118,7 +118,7 @@ static LogicalResult addLayoutsToTensorPhaseOps(
               }
               std::optional<IREE::Encoding::EncodingAttr> encodingAttr =
                   getEncodingWithNewLayouts(encodingType);
-              if (!encodingAttr.has_value()) {
+              if (!encodingAttr) {
                 return success();
               }
               rewriter.modifyOpInPlace(sizeOfOp, [&] {
@@ -149,10 +149,11 @@ struct SpecializeEncodingsPass
       return signalPassFailure();
     }
 
-    llvm::MapVector<StringRef, IREE::Stream::ExecutableOp> executableOps;
-    for (auto executableOp : moduleOp.getOps<IREE::Stream::ExecutableOp>()) {
-      executableOps[executableOp.getName()] = executableOp;
-    }
+    llvm::MapVector<StringRef, IREE::Stream::ExecutableOp> executableOp(
+        llvm::map_range(moduleOp.getOps<IREE::Stream::ExecutableOp>(),
+                        [](auto op) {
+                          return {op.getName(), op};
+                        }));
 
     IREE::Stream::ResolveLayoutAttrFn resolveLayoutAttr =
         usedDialects[0]->makeLayoutAttrResolver(moduleOp);