LLVMCPU reduction tiling rebase conflict fix

Co-authored-by: Max Dawkins <maximilian@nod-labs.com>
nod-ai · Oct 22, 2023 · 51dc391 · 51dc391
1 parent 1a149e0
commit 51dc391
Showing 1 changed file with 11 additions and 22 deletions.
diff --git a/compiler/src/iree/compiler/Codegen/LLVMCPU/Passes.cpp b/compiler/src/iree/compiler/Codegen/LLVMCPU/Passes.cpp
@@ -222,30 +222,19 @@ LogicalResult verifyDoubleTilingExpertPassPipelineConfig(
       }
     }
 
-    SmallVector<int64_t> thirdLevelTileSizes;
-    std::tie(thirdLevelTileSizes, std::ignore) =
-        tilingConfig.getVectorReductionSizes();
-    for (auto [index, tileSize] : llvm::enumerate(thirdLevelTileSizes)) {
-      if (tileSize != 0 && pLoopsSet.contains(index)) {
-        return op->emitOpError(
-                   "expected only reduction dims to be set in the third tiling "
-                   "level, got ")
-               << index << "-th tile size set";
+    if (!clEnableQuantizedMatmulReassociation) {
+      SmallVector<int64_t> thirdLevelTileSizes;
+      std::tie(thirdLevelTileSizes, std::ignore) =
+          tilingConfig.getVectorReductionSizes();
+      for (auto [index, tileSize] : llvm::enumerate(thirdLevelTileSizes)) {
+        if (tileSize != 0 && pLoopsSet.contains(index)) {
+          return op->emitOpError("expected only reduction dims to be set in "
+                                 "the third tiling "
+                                 "level, got ")
+                 << index << "-th tile size set";
+        }
       }
     }
-    // if (!clEnableQuantizedMatmulReassociation) {
-    //   SmallVector<int64_t> thirdLevelTileSizes;
-    //   std::tie(thirdLevelTileSizes, std::ignore) =
-    //       tilingConfig.getVectorReductionSizes();
-    //   for (auto [index, tileSize] : llvm::enumerate(thirdLevelTileSizes)) {
-    //     if (tileSize != 0 && pLoopsSet.contains(index)) {
-    //       return op->emitOpError("expected only reduction dims to be set in "
-    //                              "the third tiling "
-    //                              "level, got ")
-    //              << index << "-th tile size set";
-    //     }
-    //   }
-    // }
   }
 
   // Verify interchange