[Mlir-commits] [mlir] [MLIR][Vector] Add unrolling pattern for vector StepOp (PR #157752)
Erick Ochoa Lopez
llvmlistbot at llvm.org
Mon Sep 15 08:53:52 PDT 2025
================
@@ -7442,6 +7442,10 @@ void StepOp::inferResultRanges(ArrayRef<ConstantIntRanges> argRanges,
setResultRanges(getResult(), result);
}
+std::optional<SmallVector<int64_t, 4>> StepOp::getShapeForUnroll() {
+ return llvm::to_vector<4>(llvm::cast<VectorType>(getType()).getShape());
+}
+
----------------
amd-eochoalo wrote:
How is this different from the default implementation? If it isn't I recommend deleting this.
```suggestion
```
https://github.com/llvm/llvm-project/pull/157752
More information about the Mlir-commits
mailing list