gh0s7/mckernel
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

mcexec: use FLIB_NUM_PROCESS_ON_NODE when -n not specified (Fugaku specific)

Browse Source 
Change-Id: I1668fecfac692d56076dd10e6e03fbf992e323ec
This commit is contained in:
Balazs Gerofi
2020-08-12 07:27:00 +09:00
parent 542418b1fc
commit 5ffad78b87
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
executer/user/mcexec.c
View File

@ -2444,6 +2444,13 @@ int main(int argc, char **argv)
}
}
/* Fugaku: use FLIB_NUM_PROCESS_ON_NODE if -n is not specified */
if (getenv("FLIB_NUM_PROCESS_ON_NODE") && nr_processes == 0) {
nr_processes = atoi(getenv("FLIB_NUM_PROCESS_ON_NODE"));
__dprintf("%s: using FLIB_NUM_PROCESS_ON_NODE: %d\n",
__func__, nr_processes);
}
if (nr_processes > ncpu) {
fprintf(stderr, "error: nr_processes can't exceed nr. of CPUs\n");
return EINVAL;
@ -2572,9 +2579,10 @@ int main(int argc, char **argv)
desc->cpu = target_core;
desc->process_rank = process_rank;
/* Fugaku specific: Fujitsu node-local rank */
if (getenv("PLE_RANK_ON_NODE")) {
desc->process_rank = atoi(getenv("PLE_RANK_ON_NODE"));
if (getenv("FLIB_RANK_ON_NODE")) {
desc->process_rank = atoi(getenv("FLIB_RANK_ON_NODE"));
__dprintf("%s: rank: %d, target CPU: %d\n",
__func__, desc->process_rank, desc->cpu);
}