[Openmp-commits] [openmp] [OpenMP][DeviceRTL] implemented nteams-var ICV, omp_get_max_teams(), and omp_set_num_teams() (PR #71259)

Johannes Doerfert via Openmp-commits openmp-commits at lists.llvm.org
Mon Dec 4 11:09:23 PST 2023


================
@@ -628,7 +628,7 @@ uint64_t GenericKernelTy::getNumBlocks(GenericDeviceTy &GenericDevice,
 GenericDeviceTy::GenericDeviceTy(int32_t DeviceId, int32_t NumDevices,
                                  const llvm::omp::GV &OMPGridValues)
     : MemoryManager(nullptr), OMP_TeamLimit("OMP_TEAM_LIMIT"),
-      OMP_NumTeams("OMP_NUM_TEAMS"),
+      OMP_NumTeams("OMP_NUM_TEAMS_DEV_" + std::to_string(DeviceId)),
----------------
jdoerfert wrote:

you need to add the offset Plugin.getDeviceIdStartIndex(), hopefully it is already set at this point.

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


More information about the Openmp-commits mailing list