Browse Source

!10687 fix infer rank list typo

From: @gong_zi_yan
Reviewed-by: @zhunaipan,@stsuteng
Signed-off-by: @stsuteng
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
903385c5ff
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/parallel/_utils.py

+ 1
- 1
mindspore/parallel/_utils.py View File

@@ -259,7 +259,7 @@ def _infer_rank_list(train_map, predict_map=None):
logger.warning("predict_map does not contain %s", param_name)
continue
predict_layout = predict_map[param_name]
dev_num = np.array(predict_layout[0].prod())
dev_num = np.array(predict_layout[0]).prod()
# optimization pass
if _check_same_layout(train_layout, predict_layout):
dev_rank = _get_global_rank()


Loading…
Cancel
Save