For better testing: Fixed fake tasks per node to work with MPI3, added environment variable to fake num nodes.
parent
8bc9a12f80
commit
48eb1880e9
|
@ -272,6 +272,15 @@ int QueryNodeMapping(MPI_Comm comm, int print_nodemap) {
|
||||||
* tasks on node rank 0.
|
* tasks on node rank 0.
|
||||||
*/
|
*/
|
||||||
int GetNumNodes(MPI_Comm comm) {
|
int GetNumNodes(MPI_Comm comm) {
|
||||||
|
if (getenv("IOR_FAKE_NODES")){
|
||||||
|
int numNodes = atoi(getenv("IOR_FAKE_NODES"));
|
||||||
|
int rank;
|
||||||
|
MPI_Comm_rank(comm, & rank);
|
||||||
|
if(rank == 0){
|
||||||
|
printf("Fake number of node: using %d\n", numNodes);
|
||||||
|
}
|
||||||
|
return numNodes;
|
||||||
|
}
|
||||||
#if MPI_VERSION >= 3
|
#if MPI_VERSION >= 3
|
||||||
MPI_Comm shared_comm;
|
MPI_Comm shared_comm;
|
||||||
int shared_rank = 0;
|
int shared_rank = 0;
|
||||||
|
@ -338,6 +347,15 @@ int GetNumTasks(MPI_Comm comm) {
|
||||||
* method will return the same value it always has.
|
* method will return the same value it always has.
|
||||||
*/
|
*/
|
||||||
int GetNumTasksOnNode0(MPI_Comm comm) {
|
int GetNumTasksOnNode0(MPI_Comm comm) {
|
||||||
|
if (getenv("IOR_FAKE_TASK_PER_NODES")){
|
||||||
|
int tasksPerNode = atoi(getenv("IOR_FAKE_TASK_PER_NODES"));
|
||||||
|
int rank;
|
||||||
|
MPI_Comm_rank(comm, & rank);
|
||||||
|
if(rank == 0){
|
||||||
|
printf("Fake tasks per node: using %d\n", tasksPerNode);
|
||||||
|
}
|
||||||
|
return tasksPerNode;
|
||||||
|
}
|
||||||
#if MPI_VERSION >= 3
|
#if MPI_VERSION >= 3
|
||||||
MPI_Comm shared_comm;
|
MPI_Comm shared_comm;
|
||||||
int shared_rank = 0;
|
int shared_rank = 0;
|
||||||
|
@ -368,15 +386,6 @@ int GetNumTasksOnNode0(MPI_Comm comm) {
|
||||||
int size;
|
int size;
|
||||||
MPI_Comm_size(comm, & size);
|
MPI_Comm_size(comm, & size);
|
||||||
/* for debugging and testing */
|
/* for debugging and testing */
|
||||||
if (getenv("IOR_FAKE_TASK_PER_NODES")){
|
|
||||||
int tasksPerNode = atoi(getenv("IOR_FAKE_TASK_PER_NODES"));
|
|
||||||
int rank;
|
|
||||||
MPI_Comm_rank(comm, & rank);
|
|
||||||
if(rank == 0){
|
|
||||||
printf("Fake tasks per node: using %d\n", tasksPerNode);
|
|
||||||
}
|
|
||||||
return tasksPerNode;
|
|
||||||
}
|
|
||||||
char localhost[MAX_PATHLEN],
|
char localhost[MAX_PATHLEN],
|
||||||
hostname[MAX_PATHLEN];
|
hostname[MAX_PATHLEN];
|
||||||
int count = 1,
|
int count = 1,
|
||||||
|
|
Loading…
Reference in New Issue