p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!21241 modified hccl_tool.py 

Merge pull request !21241 from zong_shuai/master
This commit is contained in:
i-robot 2021-08-09 03:47:41 +00:00 committed by Gitee
parent 6b3ce4c42a e5801ed3c4
commit 1e4797a8f3
1 changed files with 3 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
model_zoo/utils/hccl_tools/hccl_tools.py
View File

@ -110,13 +110,9 @@ def main():
# construct hccn_table
device_ips: Dict[Any, Any] = {}
with open('/etc/hccn.conf', 'r') as fin:
for hccn_item in fin.readlines():
if hccn_item.strip().startswith('address_'):
device_id, device_ip = hccn_item.split('=')
device_id = device_id.split('_')[1]
device_ips[device_id] = device_ip.strip()
for device_id in device_num_list:
ret = os.popen("hccn_tool -i %d -ip -g" % device_id).readlines()
device_ips[str(device_id)] = ret[0].split(":")[1].replace('\n', '')
hccn_table = {'version': '1.0',
'server_count': '1',
'server_list': []}