Fix the list to singlton conversion

apache · Mar 9, 2020 · fc382d9 · fc382d9
1 parent 34f92a4
commit fc382d9
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/python/mxnet/kvstore/horovod.py b/python/mxnet/kvstore/horovod.py
@@ -121,16 +121,20 @@ def pushpull(self, key, value, out=None, priority=0):
         import horovod.mxnet as hvd
 
         if out is None:
-            hvd.allreduce_(value, average=False, name=key, priority=priority)
+            value = value if isinstance(value, list) else [value]
+            for v in value:
+                hvd.allreduce_(v, average=False, name=str(key), priority=priority)
         else:
-            out[:] = hvd.allreduce(value, average=False, name=key, priority=priority)
+            out = out if isinstance(out, list) else [out]
+            for o in out:
+                o[:] = hvd.allreduce(value, average=False, name=str(key), priority=priority)
 
     def set_optimizer(self, optimizer):
         pass
 
     @staticmethod
     def is_capable(capability):
-        pass
+        return False
 
     def save_optimizer_states(self, fname, dump_optimizer=False):
         pass