mirror of
https://github.com/paboyle/Grid.git
synced 2025-04-09 21:50:45 +01:00
state MPI/SLURM message only on world_rank zero
This commit is contained in:
parent
9877ed9bf8
commit
2a75516330
@ -21,22 +21,26 @@ void acceleratorInit(void)
|
|||||||
#define ENV_RANK_SLURM "SLURM_PROCID"
|
#define ENV_RANK_SLURM "SLURM_PROCID"
|
||||||
#define ENV_LOCAL_RANK_MVAPICH "MV2_COMM_WORLD_LOCAL_RANK"
|
#define ENV_LOCAL_RANK_MVAPICH "MV2_COMM_WORLD_LOCAL_RANK"
|
||||||
#define ENV_RANK_MVAPICH "MV2_COMM_WORLD_RANK"
|
#define ENV_RANK_MVAPICH "MV2_COMM_WORLD_RANK"
|
||||||
// We extract the local rank initialization using an environment variable
|
|
||||||
if ((localRankStr = getenv(ENV_LOCAL_RANK_OMPI)) != NULL) {
|
|
||||||
printf("OPENMPI detected\n");
|
|
||||||
rank = atoi(localRankStr);
|
|
||||||
} else if ((localRankStr = getenv(ENV_LOCAL_RANK_MVAPICH)) != NULL) {
|
|
||||||
printf("MVAPICH detected\n");
|
|
||||||
rank = atoi(localRankStr);
|
|
||||||
} else if ((localRankStr = getenv(ENV_LOCAL_RANK_SLURM)) != NULL) {
|
|
||||||
printf("SLURM detected\n");
|
|
||||||
rank = atoi(localRankStr);
|
|
||||||
} else {
|
|
||||||
printf("MPI version is unknown - bad things may happen\n");
|
|
||||||
}
|
|
||||||
if ((localRankStr = getenv(ENV_RANK_OMPI )) != NULL) { world_rank = atoi(localRankStr);}
|
if ((localRankStr = getenv(ENV_RANK_OMPI )) != NULL) { world_rank = atoi(localRankStr);}
|
||||||
if ((localRankStr = getenv(ENV_RANK_MVAPICH)) != NULL) { world_rank = atoi(localRankStr);}
|
if ((localRankStr = getenv(ENV_RANK_MVAPICH)) != NULL) { world_rank = atoi(localRankStr);}
|
||||||
if ((localRankStr = getenv(ENV_RANK_SLURM )) != NULL) { world_rank = atoi(localRankStr);}
|
if ((localRankStr = getenv(ENV_RANK_SLURM )) != NULL) { world_rank = atoi(localRankStr);}
|
||||||
|
// We extract the local rank initialization using an environment variable
|
||||||
|
if ((localRankStr = getenv(ENV_LOCAL_RANK_OMPI)) != NULL) {
|
||||||
|
if (!world_rank)
|
||||||
|
printf("OPENMPI detected\n");
|
||||||
|
rank = atoi(localRankStr);
|
||||||
|
} else if ((localRankStr = getenv(ENV_LOCAL_RANK_MVAPICH)) != NULL) {
|
||||||
|
if (!world_rank)
|
||||||
|
printf("MVAPICH detected\n");
|
||||||
|
rank = atoi(localRankStr);
|
||||||
|
} else if ((localRankStr = getenv(ENV_LOCAL_RANK_SLURM)) != NULL) {
|
||||||
|
if (!world_rank)
|
||||||
|
printf("SLURM detected\n");
|
||||||
|
rank = atoi(localRankStr);
|
||||||
|
} else {
|
||||||
|
if (!world_rank)
|
||||||
|
printf("MPI version is unknown - bad things may happen\n");
|
||||||
|
}
|
||||||
|
|
||||||
size_t totalDeviceMem=0;
|
size_t totalDeviceMem=0;
|
||||||
for (int i = 0; i < nDevices; i++) {
|
for (int i = 0; i < nDevices; i++) {
|
||||||
|
Loading…
x
Reference in New Issue
Block a user