Abort EquilibriumOptimizer loop just after max_use

It used to wait for the first attempt to apply an optimization after it has already crossed the threshold. Now, it signals that the optimizer should stop earlier, just after crossing that threshold.

Abort EquilibriumOptimizer loop just after max_use
82a585ac · Pascal Lamblin · e5216440 · 82a585ac
--- a/theano/gof/opt.py
+++ b/theano/gof/opt.py
@@ -1150,17 +1150,16 @@ class EquilibriumOptimizer(NavigatorOptimizer):
                    current_node = node
                    for lopt in self.local_optimizers:
                        process_count.setdefault(lopt, 0)
-                        if process_count[lopt] > max_use:
+                        lopt_change = self.process_node(env, node, lopt)
-                            max_use_abort = True
+                        if lopt_change:
-                            opt_name = (getattr(lopt, "name", None)
+                            process_count[lopt] += 1
-                                        or getattr(lopt, "__name__", None) or "")
+                            changed = True
-                        else:
+                            if process_count[lopt] > max_use:
-                            lopt_change = self.process_node(env, node, lopt)
+                                max_use_abort = True
-                            if lopt_change:
+                                opt_name = (getattr(lopt, "name", None)
-                                process_count[lopt] += 1
+                                            or getattr(lopt, "__name__", ""))
-                                changed = True
+                            if node not in env.nodes:
-                                if node not in env.nodes:
+                                break # go to next node
-                                    break # go to next node
            finally:
                self.detach_updater(env, u)
            self.detach_updater(env, u) #TODO: erase this line, it's redundant at best