summaryrefslogtreecommitdiff
path: root/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp
diff options
context:
space:
mode:
authorNishant Patel <nishant.b.patel@intel.com>2025-06-20 13:50:25 -0700
committerGitHub <noreply@github.com>2025-06-20 13:50:25 -0700
commit9c1ce31f546368d6296bc881e9f576ad25c20c73 (patch)
treec278c88d3552e2d3fda12bc50eac8dd747a013bb /mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp
parentb6445ac0c54992993b154875d6afb04eeaa13910 (diff)
[mlir][vector] Add unroll patterns for vector.load and vector.store (#143420)
This PR adds unroll patterns for vector.load and vector.store. This PR is follow up of #137558
Diffstat (limited to 'mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp')
-rw-r--r--mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp3
1 files changed, 2 insertions, 1 deletions
diff --git a/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp b/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp
index 54aa96ba89a0..71000b98fb8f 100644
--- a/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp
+++ b/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp
@@ -163,7 +163,8 @@ struct TestVectorUnrollingPatterns
.setFilterConstraint([](Operation *op) {
return success(
isa<arith::AddFOp, vector::FMAOp, vector::MultiDimReductionOp,
- vector::BroadcastOp>(op));
+ vector::BroadcastOp, vector::LoadOp, vector::StoreOp>(
+ op));
}));
populateVectorUnrollPatterns(
patterns, UnrollVectorOptions()