Skip to content

Commit 085172d

Browse files
Max191github-actions[bot]
authored andcommitted
restrict lifting of generic ops to batch_matmul
1 parent f524072 commit 085172d

File tree

1 file changed

+3
-1
lines changed
  • compiler/src/iree/compiler/GlobalOptimization

1 file changed

+3
-1
lines changed

compiler/src/iree/compiler/GlobalOptimization/Passes.cpp

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,9 @@ void buildGlobalOptimizationPassPipeline(
116116

117117
// Enable data tiling after they are in a canonical form.
118118
if (transformOptions.options.dataTiling) {
119-
mainPassManager.addPass(createLiftGenericToTransposeBatchMatmulPass());
119+
if (!clEnableQuantizedMatmulReassociation) {
120+
mainPassManager.addPass(createLiftGenericToTransposeBatchMatmulPass());
121+
}
120122
// Expand all vectors in vecmat/matvec ops into matrices for tiling.
121123
if (clEnableExpandVectors) {
122124
mainPassManager.addPass(createExpandVectorsPass());

0 commit comments

Comments
 (0)