diff --git a/mlir/lib/TableGen/Pattern.cpp b/mlir/lib/TableGen/Pattern.cpp --- a/mlir/lib/TableGen/Pattern.cpp +++ b/mlir/lib/TableGen/Pattern.cpp @@ -345,7 +345,7 @@ case Kind::Value: { assert(index < 0 && "only allowed for symbol bound to result"); assert(op == nullptr); - auto repl = formatv(fmt, formatv("{{{0}}", name)); + auto repl = formatv(fmt, formatv("ValueRange{{{0}}", name)); LLVM_DEBUG(llvm::dbgs() << repl << " (Value)\n"); return std::string(repl); } diff --git a/mlir/tools/mlir-tblgen/RewriterGen.cpp b/mlir/tools/mlir-tblgen/RewriterGen.cpp --- a/mlir/tools/mlir-tblgen/RewriterGen.cpp +++ b/mlir/tools/mlir-tblgen/RewriterGen.cpp @@ -612,15 +612,8 @@ auto val = handleResultPattern(resultTree, offsets[i], 0); os.indent(4) << "\n"; // Resolve each symbol for all range use so that we can loop over them. - // We need an explicit cast to `SmallVector` to capture the cases where - // `{0}` resolves to an `Operation::result_range` as well as cases that - // are not iterable (e.g. vector that gets wrapped in additional braces by - // RewriterGen). - // TODO(b/147096809): Revisit the need for materializing a vector. os << symbolInfoMap.getAllRangeUse( - val, - " for (auto v : SmallVector{ {0} }) {{ " - "tblgen_repl_values.push_back(v); }", + val, " for (Value v : {0}) {{ tblgen_repl_values.push_back(v); }", "\n"); } os.indent(4) << "\n";