Add another pattern set for vectorization

nod-ai · Aug 29, 2024 · 6c22541 · 6c22541
1 parent 603bd5d
commit 6c22541
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/compiler/plugins/target/AMD-AIE/iree-amd-aie/Transforms/AMDAIEVectorization.cpp b/compiler/plugins/target/AMD-AIE/iree-amd-aie/Transforms/AMDAIEVectorization.cpp
@@ -100,6 +100,7 @@ void AMDAIEVectorizationPass::runOnOperation() {
   RewritePatternSet vectorizationPatterns(funcOp.getContext());
 
   vector::populateVectorReductionToContractPatterns(vectorizationPatterns);
+  vector::populateSinkVectorOpsPatterns(vectorizationPatterns);
 
   // Including this pattern prevents broadcasting in vector.transfer_read ops
   vector::populateVectorTransferPermutationMapLoweringPatterns(