Bug: Executor - fix bug in result writing to files for mpi mode (#328)

**Description** fix the bug in result writing to files for mpi mode.

Bug: Executor - fix bug in result writing to files for mpi mode (#328)
**Description** fix the bug in result writing to files for mpi mode.
84359fd8 · user4543 · GitHub · b3c95f18 · 84359fd8
Unverified Commit 84359fd8 authored Mar 16, 2022 by user4543 Committed by GitHub Mar 15, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

superbench/executor/executor.py superbench/executor/executor.py +1 -1

No files found.
--- a/superbench/executor/executor.py
+++ b/superbench/executor/executor.py
@@ -141,7 +141,7 @@ def __get_rank_id(self):
        Return:
            int: Rank ID.
        """
-        for rank_env in ['PROC_RANK', 'LOCAL_RANK']:
+        for rank_env in ['PROC_RANK', 'LOCAL_RANK', 'OMPI_COMM_WORLD_LOCAL_RANK']:
            if os.getenv(rank_env):
                return int(os.getenv(rank_env))