From 6c22541b6210add7fa9f250b7bd40caa2551eca1 Mon Sep 17 00:00:00 2001 From: Abhishek Varma Date: Thu, 29 Aug 2024 05:39:56 +0000 Subject: [PATCH] Add another pattern set for vectorization --- .../AMD-AIE/iree-amd-aie/Transforms/AMDAIEVectorization.cpp | 1 + 1 file changed, 1 insertion(+) 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 index fa661b784..cec0c635b 100644 --- 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(