[llvm] [mlir] [MLIR][OpenMP] Add MLIR Lowering Support for dist_schedule (PR #152736)

Tom Eccles via llvm-commits llvm-commits at lists.llvm.org
Mon Aug 11 03:06:55 PDT 2025


================
@@ -4783,13 +4843,18 @@ OpenMPIRBuilder::applyStaticChunkedWorkshareLoop(DebugLoc DL,
   Builder.SetCurrentDebugLocation(DL);
 
   // TODO: Detect overflow in ubsan or max-out with current tripcount.
-  Value *CastedChunkSize =
-      Builder.CreateZExtOrTrunc(ChunkSize, InternalIVTy, "chunksize");
+  Value *CastedChunkSize = Builder.CreateZExtOrTrunc(
+      ChunkSize ? ChunkSize : Zero, InternalIVTy, "chunksize");
+  Value *CastestDistScheduleChunkSize = Builder.CreateZExtOrTrunc(
----------------
tblah wrote:

```suggestion
  Value *CastedDistScheduleChunkSize = Builder.CreateZExtOrTrunc(
```

https://github.com/llvm/llvm-project/pull/152736


More information about the llvm-commits mailing list