From 53287675a447f430fb975d42ff3266f60bc6d76f Mon Sep 17 00:00:00 2001
From: Krzysztof Drewniak <Krzysztof.Drewniak@amd.com>
Date: Mon, 18 Nov 2024 19:06:21 +0000
Subject: [PATCH] Review feedback

---
 .../iree/compiler/Codegen/Common/GPU/GPUDistributeForall.cpp  | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/compiler/src/iree/compiler/Codegen/Common/GPU/GPUDistributeForall.cpp b/compiler/src/iree/compiler/Codegen/Common/GPU/GPUDistributeForall.cpp
index b9d6e6c09e3c5..334427cfffb97 100644
--- a/compiler/src/iree/compiler/Codegen/Common/GPU/GPUDistributeForall.cpp
+++ b/compiler/src/iree/compiler/Codegen/Common/GPU/GPUDistributeForall.cpp
@@ -88,6 +88,10 @@ LogicalResult resolveGPUMappedForallOp(RewriterBase &rewriter,
   assert(!(hasThreadMapping && hasWarpMapping));
   Value flatId = linearThreadId;
   if (hasWarpMapping) {
+    if (flatWorkgroupSize % subgroupSize != 0) {
+      return forallOp->emitOpError(
+          "found warp mapped forall with non-multiple workgroup size");
+    }
     flatId = rewriter
                  .create<affine::AffineDelinearizeIndexOp>(
                      loc, flatId,