commit e363ef92d09b95e3462fa34ee4e8c4e517f2069e
author Dominik Adamski <dominik.adamski@amd.com> Fri Sep 26 13:57:51 2025 +0200
committer Copybara-Service <copybara-worker@google.com> Fri Sep 26 05:00:18 2025 -0700
tree 693c333565c78d51018ebba37885f755b9e45158
parent 445171b5ee8c1186aec2cd3021696e49e79fe691

[Flang][OpenMP] Enable no-loop kernels (#155818)

Enable the generation of no-loop kernels for Fortran OpenMP code. target
teams distribute parallel do pragmas can be promoted to no-loop kernels
if the user adds the -fopenmp-assume-teams-oversubscription and
-fopenmp-assume-threads-oversubscription flags.

If the OpenMP kernel contains reduction or num_teams clauses, it is not
promoted to no-loop mode.

The global OpenMP device RTL oversubscription flags no longer force
no-loop code generation for Fortran.

GitOrigin-RevId: 83ef38a274b76c8877f15d3758d955772f91cf2d

device/src/Workshare.cpp

diff --git a/device/src/Workshare.cpp b/device/src/Workshare.cpp
index 59a2cc3..653104c 100644
--- a/device/src/Workshare.cpp
+++ b/device/src/Workshare.cpp
@@ -800,10 +800,6 @@
 
     // If we know we have more threads than iterations we can indicate that to
     // avoid an outer loop.
-    if (config::getAssumeThreadsOversubscription()) {
-      OneIterationPerThread = true;
-    }
-
     if (OneIterationPerThread)
       ASSERT(NumThreads >= NumIters, "Broken assumption");
 
@@ -851,10 +847,6 @@
 
     // If we know we have more blocks than iterations we can indicate that to
     // avoid an outer loop.
-    if (config::getAssumeTeamsOversubscription()) {
-      OneIterationPerThread = true;
-    }
-
     if (OneIterationPerThread)
       ASSERT(NumBlocks >= NumIters, "Broken assumption");
 
@@ -914,11 +906,6 @@
 
     // If we know we have more threads (across all blocks) than iterations we
     // can indicate that to avoid an outer loop.
-    if (config::getAssumeTeamsOversubscription() &
-        config::getAssumeThreadsOversubscription()) {
-      OneIterationPerThread = true;
-    }
-
     if (OneIterationPerThread)
       ASSERT(NumBlocks * NumThreads >= NumIters, "Broken assumption");