diff options
| author | Nishant Patel <nishant.b.patel@intel.com> | 2025-09-16 15:33:08 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-09-16 15:33:08 -0700 |
| commit | 0e5c32bd6d305f417d1b4fc023616711206eccc4 (patch) | |
| tree | bee5381bc2809bb37a3f7288810ef68e9214ffe9 /mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp | |
| parent | 4ab8dabc252f802134bfea6193f9a274f0bdc143 (diff) | |
[MLIR][Vector] Add unrolling pattern for vector StepOp (#157752)
This PR adds unrolling pattern for vector.step op to VectorUnroll
transform.
Diffstat (limited to 'mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp')
| -rw-r--r-- | mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp b/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp index 3360f3b70a81..72dd103b33f7 100644 --- a/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp +++ b/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp @@ -172,6 +172,12 @@ struct TestVectorUnrollingPatterns .setFilterConstraint([](Operation *op) { return success(isa<vector::ReductionOp>(op)); })); + populateVectorUnrollPatterns(patterns, + UnrollVectorOptions() + .setNativeShape(ArrayRef<int64_t>{8}) + .setFilterConstraint([](Operation *op) { + return success(isa<vector::StepOp>(op)); + })); populateVectorUnrollPatterns( patterns, UnrollVectorOptions() .setNativeShape(ArrayRef<int64_t>{1, 3, 4, 2}) |