Set MPI threading mode for hyd-7c machine

[ROCm/rocshmem commit: 8b14d9f07d]
This commit is contained in:
Brandon Potter
2024-07-11 07:29:44 -07:00
parent 3465c887d4
commit 0d9e64d446
4 ha cambiato i file con 6 aggiunte e 6 eliminazioni
@@ -343,7 +343,7 @@ void GPUIBBackend::init_mpi_once(MPI_Comm comm) {
if (init_done == 0) {
int provided;
NET_CHECK(
MPI_Init_thread(nullptr, nullptr, MPI_THREAD_SINGLE, &provided));
MPI_Init_thread(nullptr, nullptr, MPI_THREAD_MULTIPLE, &provided));
}
if (comm == MPI_COMM_NULL) {
NET_CHECK(MPI_Comm_dup(MPI_COMM_WORLD, &backend_comm));
@@ -55,7 +55,7 @@ class CommunicatorMPI {
MPI_Initialized(&initialized);
if (!initialized) {
int provided;
MPI_Init_thread(nullptr, nullptr, MPI_THREAD_SINGLE, &provided);
MPI_Init_thread(nullptr, nullptr, MPI_THREAD_MULTIPLE, &provided);
}
MPI_Comm_rank(comm_, &my_pe_);
MPI_Comm_size(comm_, &num_pes_);
@@ -31,7 +31,7 @@ MPIInitSingleton::MPIInitSingleton() {
if (!pre_init_done) {
int provided;
MPI_Init_thread(nullptr, nullptr, MPI_THREAD_SINGLE, &provided);
MPI_Init_thread(nullptr, nullptr, MPI_THREAD_MULTIPLE, &provided);
}
MPI_Comm_size(MPI_COMM_WORLD, &nprocs_);
@@ -48,9 +48,9 @@ MPITransport::MPITransport(MPI_Comm comm, Queue* q)
int provided{};
if (!init_done) {
NET_CHECK(MPI_Init_thread(0, 0, MPI_THREAD_SINGLE, &provided));
if (provided != MPI_THREAD_SINGLE) {
std::cerr << "MPI_THREAD_SINGLE support disabled.\n";
NET_CHECK(MPI_Init_thread(0, 0, MPI_THREAD_MULTIPLE, &provided));
if (provided != MPI_THREAD_MULTIPLE) {
std::cerr << "MPI_THREAD_MULTIPLE support disabled.\n";
}
}
if (comm == MPI_COMM_NULL) comm = MPI_COMM_WORLD;