[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