[Mlir-commits] [mlir] [mlir][loops] Add getters for multi dim loop variables in `LoopLikeOpInterface` (PR #94516)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Fri Jun 7 08:22:27 PDT 2024
================
@@ -846,8 +850,11 @@ def ParallelOp : SCF_Op<"parallel",
];
let extraClassDeclaration = [{
- ValueRange getInductionVars() {
- return getBody()->getArguments();
+ // Get induction variables.
+ SmallVector<Value> getInductionVars() {
+ auto maybeInductionVars = getLoopInductionVars();;
----------------
srcarroll wrote:
oh weird. i didn't know that even compiles
https://github.com/llvm/llvm-project/pull/94516
More information about the Mlir-commits
mailing list