diff --git a/mlir/include/mlir/Dialect/Affine/IR/AffineOps.td b/mlir/include/mlir/Dialect/Affine/IR/AffineOps.td --- a/mlir/include/mlir/Dialect/Affine/IR/AffineOps.td +++ b/mlir/include/mlir/Dialect/Affine/IR/AffineOps.td @@ -233,7 +233,7 @@ static StringRef getLowerBoundAttrName() { return "lower_bound"; } static StringRef getUpperBoundAttrName() { return "upper_bound"; } - Value getInductionVar() { return getBody()->getArgument(0); } + BlockArgument getInductionVar() { return getBody()->getArgument(0); } Block::BlockArgListType getRegionIterArgs() { return getBody()->getArguments().drop_front(); }