[mlir] Add materializeOpFoldResults to turn OpFoldResult array into v…

…alues. Differential Revision: https://reviews.llvm.org/D133346
llvm · Sep 6, 2022 · c141c03 · c141c03
1 parent 4ce3848
commit c141c03
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/mlir/include/mlir/Dialect/Linalg/Utils/Utils.h b/mlir/include/mlir/Dialect/Linalg/Utils/Utils.h
@@ -193,6 +193,8 @@ Value materializeOpFoldResult(ImplicitLocOpBuilder &builder,
                               OpFoldResult opFoldResult);
 Value materializeOpFoldResult(OpBuilder &b, Location loc,
                               OpFoldResult opFoldResult);
+Value materializeOpFoldResult(OpBuilder &b, Location loc,
+                              ArrayRef<OpFoldResult> opFoldResults);
 
 /// A struct containg offsets-sizes-strides arguments of the tiled shape.
 struct SliceParameters {

diff --git a/mlir/lib/Dialect/Linalg/Utils/Utils.cpp b/mlir/lib/Dialect/Linalg/Utils/Utils.cpp
@@ -946,6 +946,16 @@ Value materializeOpFoldResult(OpBuilder &builder, Location loc,
   return materializeOpFoldResult(b, opFoldResult);
 }
 
+SmallVector<Value>
+materializeOpFoldResults(OpBuilder &builder, Location loc,
+                         ArrayRef<OpFoldResult> opFoldResults) {
+  ImplicitLocOpBuilder b(loc, builder);
+  SmallVector<Value> values;
+  for (const auto &opFoldResult : opFoldResults)
+    values.push_back(materializeOpFoldResult(b, opFoldResult));
+  return values;
+}
+
 SmallVector<Optional<SliceParameters>>
 computeAllSliceParameters(OpBuilder &builder, Location loc, LinalgOp linalgOp,
                           ValueRange valuesToTile, ArrayRef<OpFoldResult> ivs,