From 842c62669a92c0d55841aa2c84a41fc45022c5f4 Mon Sep 17 00:00:00 2001
From: Bangtian Liu <liubangtian@gmail.com>
Date: Thu, 11 Jul 2024 15:03:46 -0500
Subject: [PATCH] add a check to return failure

Signed-off-by: Bangtian Liu <liubangtian@gmail.com>
---
 .../iree/compiler/Codegen/Common/GPU/WorkgroupReordering.cpp    | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/compiler/src/iree/compiler/Codegen/Common/GPU/WorkgroupReordering.cpp b/compiler/src/iree/compiler/Codegen/Common/GPU/WorkgroupReordering.cpp
index c7f2e8692c7a..d9a5f961566b 100644
--- 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);