From 551cc701c6c6450a370f243aeea9d8d858da5a0c Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Tue, 13 Jul 2021 18:51:53 +0800
Subject: [PATCH] docs(distributed.functional): add return type for
 all_reduce_max (jira #MGE-2706)

GitOrigin-RevId: a29f1f1880674a45fa593582b34b807be1e46786
---
 imperative/python/megengine/distributed/functional.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/imperative/python/megengine/distributed/functional.py b/imperative/python/megengine/distributed/functional.py
index 9171b64e..f0650b6f 100644
--- a/imperative/python/megengine/distributed/functional.py
+++ b/imperative/python/megengine/distributed/functional.py
@@ -271,6 +271,7 @@ def all_reduce_max(
     :param inp: input tensor.
     :param group: communication group.
     :param device: execution device.
+    :returns: reduced tensor. 
     """
     mode = CollectiveComm.Mode.ALL_REDUCE_MAX
     return collective_comm(inp, mode, group, device)