[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