OpenI
/
MegEngine
Not watched
1
0
Fork 0
Code
Releases 31 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
Browse Source

docs(distributed.functional): add return type for all_reduce_max (jira #MGE-2706) 

GitOrigin-RevId: a29f1f1880
release-1.5
Megvii Engine Team huangxinda 3 years ago
parent
72ff7aeccb
commit
551cc701c6
1 changed files with 1 additions and 0 deletions
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -0
      imperative/python/megengine/distributed/functional.py

+ 1
- 0
imperative/python/megengine/distributed/functional.py View File

@@ -271,6 +271,7 @@ def all_reduce_max(
:param inp: input tensor. :param inp: input tensor.
:param group: communication group. :param group: communication group.
:param device: execution device. :param device: execution device.
:returns: reduced tensor.
""" """
mode = CollectiveComm.Mode.ALL_REDUCE_MAX mode = CollectiveComm.Mode.ALL_REDUCE_MAX
return collective_comm(inp, mode, group, device) return collective_comm(inp, mode, group, device)


Write Preview
Loading…
Cancel
Save