|
|
|
@@ -91,11 +91,12 @@ run_ascend(){ |
|
|
|
#device_num=${#device_id_list} |
|
|
|
cat $2 | awk -F "[device_id]" '/device_id/{print$0}' >temp.log |
|
|
|
array=$(cat temp.log | awk -F "[:]" '/device_id/{print$2}') |
|
|
|
rm temp.log |
|
|
|
|
|
|
|
IFS=" " read -ra device_list <<<$array |
|
|
|
first_device=${device_list[0]:1:1} |
|
|
|
device_num=${#device_list[*]} |
|
|
|
|
|
|
|
#device_num=${#device_list[*]} |
|
|
|
device_num=$(cat temp.log | wc -l) |
|
|
|
rm temp.log |
|
|
|
ulimit -u unlimited |
|
|
|
export DEVICE_NUM=${device_num} |
|
|
|
export RANK_SIZE=${device_num} |
|
|
|
|