From 56318d9c16ca600b7994b4c0aecf3fc453896eb3 Mon Sep 17 00:00:00 2001 From: Ziyan Date: Mon, 28 Dec 2020 12:48:14 +0800 Subject: [PATCH] fix infer rank list typo --- mindspore/parallel/_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/parallel/_utils.py b/mindspore/parallel/_utils.py index 0a1b9f2328..e602fa6a3a 100644 --- a/mindspore/parallel/_utils.py +++ b/mindspore/parallel/_utils.py @@ -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()