restrict lifting of generic ops to batch_matmul

nod-ai · Dec 1, 2023 · b5867e6 · b5867e6
1 parent 40d58f0
commit b5867e6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/compiler/src/iree/compiler/GlobalOptimization/Passes.cpp b/compiler/src/iree/compiler/GlobalOptimization/Passes.cpp
@@ -95,7 +95,9 @@ void buildGlobalOptimizationPassPipeline(
 
   // Enable data tiling after they are in a canonical form.
   if (transformOptions.options.dataTiling) {
-    mainPassManager.addPass(createLiftGenericToTransposeBatchMatmulPass());
+    if (!clEnableQuantizedMatmulReassociation) {
+      mainPassManager.addPass(createLiftGenericToTransposeBatchMatmulPass());
+    }
     // Expand all vectors in vecmat/matvec ops into matrices for tiling.
     mainPassManager.addPass(createExpandVectorsPass());
     mainPassManager.addPass(createSetEncodingPass());