From e5801ed3c4c58e985a19f55fffa1a337035988cf Mon Sep 17 00:00:00 2001 From: zong_shuai Date: Mon, 2 Aug 2021 15:27:45 +0800 Subject: [PATCH] modified hccl_tools.py modified hccl_tools.py --- model_zoo/utils/hccl_tools/hccl_tools.py | 10 +++------- 1 file changed, 3 insertions(+), 7 deletions(-) diff --git a/model_zoo/utils/hccl_tools/hccl_tools.py b/model_zoo/utils/hccl_tools/hccl_tools.py index f019f179bd8..2df333b5efc 100644 --- a/model_zoo/utils/hccl_tools/hccl_tools.py +++ b/model_zoo/utils/hccl_tools/hccl_tools.py @@ -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': []}