Browse Source

!16053 add detailed describtion of communication initialization log in cpu

From: @yao_yf
Reviewed-by: @stsuteng,@yangzhenzhang
Signed-off-by: @stsuteng
pull/16053/MERGE
mindspore-ci-bot Gitee 4 years ago
parent
commit
2c5c249ec1
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      mindspore/communication/management.py

+ 2
- 1
mindspore/communication/management.py View File

@@ -83,7 +83,8 @@ def init(backend_name=None):
elif device_target == "GPU":
backend_name = "nccl"
else:
raise RuntimeError("Device target {} is not supported.".format(device_target))
raise RuntimeError("Device target {} is not supported in parallel initialization, "
"please use Ascend or GPU.".format(device_target))
if not isinstance(backend_name, str):
raise TypeError("Backend name must be a string, but got {}".format(type(backend_name)))



Loading…
Cancel
Save