[Mlir-commits] [mlir] [mlir][loops] Add getters for multi dim loop variables in `LoopLikeOpInterface` (PR #94516)

Ivan Butygin llvmlistbot at llvm.org
Wed Jun 5 11:55:09 PDT 2024


================
@@ -235,6 +231,35 @@ def LoopLikeOpInterface : OpInterface<"LoopLikeOpInterface"> {
   }];
 
   let extraSharedClassDeclaration = [{
+    /// If there is a single induction variable return it, otherwise return
+    /// std::nullopt.
+    ::std::optional<::mlir::Value> getSingleInductionVar() {
+      if (this->getInductionVars().size() == 1)
----------------
Hardcode84 wrote:

this code will create a temporary `SmallVector` just to get size and then another one on the next line, better to cache.

https://github.com/llvm/llvm-project/pull/94516


More information about the Mlir-commits mailing list