Browse Source

add detailed description communication init log in cpu

pull/16053/head
yao_yf 4 years ago
parent
commit
a31cd2ff47
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