diff --git a/compiler/src/iree/compiler/Codegen/LLVMCPU/KernelDispatch.cpp b/compiler/src/iree/compiler/Codegen/LLVMCPU/KernelDispatch.cpp index 6c33ad9956344..1030dbfb82ea6 100644 --- a/compiler/src/iree/compiler/Codegen/LLVMCPU/KernelDispatch.cpp +++ b/compiler/src/iree/compiler/Codegen/LLVMCPU/KernelDispatch.cpp @@ -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; }