Speed up group executor (apache#16069)

* Speed up group executor Current implementation is O(n^2), this implementation is O(n) * Speed up group executor Current implementation is O(n^2), this implementation is O(n) * CI
gyshi · Sep 7, 2019 · 9825de6 · 9825de6
1 parent 52ab212
commit 9825de6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/mxnet/module/executor_group.py b/python/mxnet/module/executor_group.py
@@ -273,9 +273,9 @@ def __init__(self, symbol, contexts, workload, data_shapes, label_shapes, param_
         self.data_layouts = None
         self.label_layouts = None
         self.output_names = self.symbol.list_outputs()
-        self.output_layouts = [DataDesc.get_batch_axis(self.symbol[name].attr('__layout__'))
-                               for name in self.output_names]
-        self.num_outputs = len(self.symbol.list_outputs())
+        self.num_outputs = len(self.output_names)
+        self.output_layouts = [DataDesc.get_batch_axis(self.symbol[index].attr('__layout__'))
+                               for index in range(self.num_outputs)]
 
         self.bind_exec(data_shapes, label_shapes, shared_group)