Set MPI threading mode for hyd-7c machine
[ROCm/rocshmem commit: 8b14d9f07d]
This commit is contained in:
@@ -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;
|
||||
|
||||
Fai riferimento in un nuovo problema
Block a user