Skip to content

Commit

Permalink
Check LHS batch pack
Browse files Browse the repository at this point in the history
  • Loading branch information
Jerry Wu committed Aug 29, 2023
1 parent a0da52a commit a170aa5
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
4 changes: 4 additions & 0 deletions compiler/src/iree/compiler/Codegen/LLVMCPU/KernelDispatch.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -1236,6 +1236,10 @@ getDefaultDistributionTileSizes(TilingInterface op) {
}

static bool isPackMatmulLHS(tensor::PackOp op) {
if (op.getSourceRank() == 3 && op.getInnerDimsPos().size() == 2 &&
op.getInnerDimsPos()[0] == 1 && op.getInnerDimsPos()[1] == 2) {
return true;
}
return op.getSourceRank() == 2 && op.getInnerDimsPos().size() == 2 &&
op.getInnerDimsPos()[0] == 0 && op.getInnerDimsPos()[1] == 1;
}
Expand Down
4 changes: 0 additions & 4 deletions compiler/src/iree/compiler/Codegen/LLVMCPU/Passes.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -513,10 +513,6 @@ void addMultiTilingExpertPassPipeline(

addBufferizePasses(nestedModulePM);

// Perform memref-based transfer_read/write optimizations.
nestedModulePM.addNestedPass<func::FuncOp>(
createOptimizeVectorTransferPass(/*flatten=*/false));

// Run IREE specific passes before vector lowering expert.
nestedModulePM.addNestedPass<func::FuncOp>(
createRemoveSingleIterationLoopPass());
Expand Down

0 comments on commit a170aa5

Please sign in to comment.