[clang] [llvm] [OpenMP][amdgpu] Use max teams for amdgpu-max-num-workgroups (PR #195203)

Shilei Tian via cfe-commits cfe-commits at lists.llvm.org
Fri May 1 11:26:24 PDT 2026


================
@@ -8349,11 +8074,12 @@ OpenMPIRBuilder::readTeamBoundsForKernel(const Triple &, Function &Kernel) {
 
 void OpenMPIRBuilder::writeTeamsForKernel(const Triple &T, Function &Kernel,
                                           int32_t LB, int32_t UB) {
-  if (T.isNVPTX())
-    if (UB > 0)
+  if (UB > 0) {
+    if (T.isNVPTX())
       Kernel.addFnAttr(NVVMAttr::MaxClusterRank, llvm::utostr(UB));
-  if (T.isAMDGPU())
-    Kernel.addFnAttr("amdgpu-max-num-workgroups", llvm::utostr(LB) + ",1,1");
+    if (T.isAMDGPU())
+      Kernel.addFnAttr("amdgpu-max-num-workgroups", llvm::utostr(LB) + ",1,1");
+  }
 
   Kernel.addFnAttr("omp_target_num_teams", std::to_string(LB));
----------------
shiltian wrote:

Can you help me to refresh my memory on why this is a range?

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


More information about the cfe-commits mailing list