add a check to return failure

Signed-off-by: Bangtian Liu <liubangtian@gmail.com>
iree-org · Jul 11, 2024 · 842c626 · 842c626
1 parent baf9997
commit 842c626
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/compiler/src/iree/compiler/Codegen/Common/GPU/WorkgroupReordering.cpp b/compiler/src/iree/compiler/Codegen/Common/GPU/WorkgroupReordering.cpp
@@ -283,6 +283,8 @@ static LogicalResult reorderWorkgroupsInFunc(FunctionOpInterface funcOp,
         makeSwizzledIds(funcOp.getLoc(), builder, workgroupIdX, workgroupIdY,
                         workgroupCntX, workgroupCntY, reorderWgTileSize);
   } else if (strategy == ReorderWorkgroupsStrategy::ChipletGroup) {
+    if (numXCDs <= 1)
+      return failure();
     std::tie(newWorkgroupIdX, newWorkgroupIdY) = makeChipletGroupedIds(
         funcOp.getLoc(), builder, workgroupIdX, workgroupIdY, workgroupCntX,
         workgroupCntY, reorderWgTileSize, numXCDs);