Unverified Commit 84359fd8 authored by user4543's avatar user4543 Committed by GitHub
Browse files

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.
parent b3c95f18
...@@ -141,7 +141,7 @@ def __get_rank_id(self): ...@@ -141,7 +141,7 @@ def __get_rank_id(self):
Return: Return:
int: Rank ID. 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): if os.getenv(rank_env):
return int(os.getenv(rank_env)) return int(os.getenv(rank_env))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment