Change AffineApplyOp to produce a single result, simplifying the code that

works with it, and updating the g3docs.

PiperOrigin-RevId: 231120927

1 files changed, 3 insertions, 4 deletions

diff --git a/mlir/lib/Transforms/MaterializeVectors.cpp b/mlir/lib/Transforms/MaterializeVectors.cpp
index 7dd3cecdfed..4360a4e1b96 100644
--- a/mlir/lib/Transforms/MaterializeVectors.cpp
+++ b/mlir/lib/Transforms/MaterializeVectors.cpp
@@ -376,11 +376,10 @@ reindexAffineIndices(FuncBuilder *b, VectorType hwVectorType,
 
   // Create a bunch of single result maps.
   return functional::map(
-      [b, numIndices, memrefIndices](AffineExpr expr) {
+      [b, numIndices, memrefIndices](AffineExpr expr) -> Value * {
         auto map = AffineMap::get(numIndices, 0, expr, {});
-        auto app = makeComposedAffineApply(b, b->getInsertionPoint()->getLoc(),
-                                           map, memrefIndices);
-        return app->getResult(0);
+        return makeComposedAffineApply(b, b->getInsertionPoint()->getLoc(), map,
+                                       memrefIndices);
       },
       affineExprs);
 }