[llvm] [Offload][OMPX] Add the runtime support for multi-dim grid and block (PR #118042)

Joseph Huber via llvm-commits llvm-commits at lists.llvm.org
Fri Nov 29 08:18:01 PST 2024


================
@@ -526,20 +526,21 @@ GenericKernelTy::getKernelLaunchEnvironment(
 
 Error GenericKernelTy::printLaunchInfo(GenericDeviceTy &GenericDevice,
                                        KernelArgsTy &KernelArgs,
-                                       uint32_t NumThreads,
-                                       uint64_t NumBlocks) const {
-  INFO(OMP_INFOTYPE_PLUGIN_KERNEL, GenericDevice.getDeviceId(),
-       "Launching kernel %s with %" PRIu64
-       " blocks and %d threads in %s mode\n",
-       getName(), NumBlocks, NumThreads, getExecutionModeName());
+                                       uint32_t NumThreads[3],
+                                       uint32_t NumBlocks[3]) const {
+  if (!IsBareKernel) {
----------------
jhuber6 wrote:

We could make this print unconditionally and only list the y, z dimensions if they're non-zero.

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


More information about the llvm-commits mailing list