未验证 提交 c5a449f9 编写于 作者: J Jeff Rasley 提交者: GitHub

Update launcher to set local rank environ variable (#597)

* Update launch.py

* formatting
上级 a4763f55
......@@ -113,6 +113,7 @@ def main():
# each process's rank
dist_rank = global_rank_mapping[local_node][local_rank]
current_env["RANK"] = str(dist_rank)
current_env["LOCAL_RANK"] = str(local_rank)
# spawn the processes
cmd = [
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册