Review feedback

iree-org · Nov 18, 2024 · 5328767 · 5328767
1 parent 47b71e4
commit 5328767
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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,