diff options
Diffstat (limited to 'mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp')
| -rw-r--r-- | mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp | 3 |
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() |