Skip to content

Commit

Permalink
Modify warning messages about number of GPUs (#4315)
Browse files Browse the repository at this point in the history
  • Loading branch information
WeiqunZhang authored Jan 28, 2025
1 parent 2f3a8c7 commit 12d0882
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions Src/Base/AMReX_GpuDevice.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -204,7 +204,7 @@ Device::Initialize ()
}
else {
if (amrex::Verbose() && ParallelDescriptor::IOProcessor()) {
amrex::Warning("Multiple GPUs are visible to each MPI rank, This may lead to incorrect or suboptimal rank-to-GPU mapping.");
amrex::Warning("Multiple GPUs are visible to each MPI rank. This is usually not an issue. But this may lead to incorrect or suboptimal rank-to-GPU mapping.");
}
if (ParallelDescriptor::NProcsPerNode() == gpu_device_count) {
device_id = ParallelDescriptor::MyRankInNode();
Expand Down Expand Up @@ -322,7 +322,8 @@ Device::Initialize ()
<< ((num_devices_used == 1) ? " device.\n"
: " devices.\n");
if (num_devices_used < ParallelDescriptor::NProcs() && ParallelDescriptor::IOProcessor()) {
amrex::Warning("There are more MPI processes than the number of GPUs.");
amrex::Warning("There are more MPI processes than the number of unique GPU devices. This is not necessarily a problem.\n"
"For example this could happen when a device such as MI300A is partitioned into multiple subdevices.");
}
}

Expand Down

0 comments on commit 12d0882

Please sign in to comment.