[Mlir-commits] [mlir] [mlir] Add insert method for ApplyToEachResultList class (PR #109582)

[llvmlistbot at llvm.org]

https://github.com/fawdlstty created https://github.com/llvm/llvm-project/pull/109582

#94968

>From 38ff2e69ec736a942688385806ba32c92a7815b4 Mon Sep 17 00:00:00 2001
From: fawdlstty <f at fawdlstty.com>
Date: Mon, 23 Sep 2024 00:48:13 +0800
Subject: [PATCH] add insert method

---
 .../mlir/Dialect/Transform/Interfaces/TransformInterfaces.h | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/mlir/include/mlir/Dialect/Transform/Interfaces/TransformInterfaces.h b/mlir/include/mlir/Dialect/Transform/Interfaces/TransformInterfaces.h
index 43193e4cd4cf63..728a99e4f67661 100644
--- a/mlir/include/mlir/Dialect/Transform/Interfaces/TransformInterfaces.h
+++ b/mlir/include/mlir/Dialect/Transform/Interfaces/TransformInterfaces.h
@@ -1432,6 +1432,12 @@ class ApplyToEachResultList {
   void push_back(Attribute attr) { results.push_back(attr); }
   void push_back(ApplyToEachResult r) { results.push_back(r); }
 
+  // Inserts an element to the list.
+  using iterator = SmallVector<ApplyToEachResult>::iterator;
+  void insert(iterator I, Operation *op) { results.insert(I, op); }
+  void insert(iterator I, Attribute attr) { results.insert(I, attr); }
+  void insert(iterator I, ApplyToEachResult r) { results.insert(I, r); }
+
   /// Reserves space for `size` elements in the list.
   void reserve(unsigned size) { results.reserve(size); }