In FirstReady node manager, use node name as tie-breaker when multiple nodes

have same time_ready.

PiperOrigin-RevId: 179940344

1 files changed, 8 insertions, 3 deletions

diff --git a/tensorflow/core/grappler/costs/virtual_scheduler.cc b/tensorflow/core/grappler/costs/virtual_scheduler.cc
index f74ec8cb9f..fce50e33d6 100644
--- a/tensorflow/core/grappler/costs/virtual_scheduler.cc
+++ b/tensorflow/core/grappler/costs/virtual_scheduler.cc
@@ -118,9 +118,14 @@ FirstReadyManager::FirstReadyManager(
     : ReadyNodeManager(), node_state_(node_state) {
   std::make_heap(nodes_.begin(), nodes_.end());
   greater_ = [this](const NodeDef* a, const NodeDef* b) -> bool {
-    // Note: we need a node with minimum time_ready, not
-    // maximum; hence, using a > b for comparison function.
-    return node_state_->at(a).time_ready > node_state_->at(b).time_ready;
+    if (node_state_->at(a).time_ready == node_state_->at(b).time_ready) {
+      // Use Node name as tie-breaker for deterministic node scheduling.
+      return a->name().compare(b->name()) > 0;
+    } else {
+      // Note: we need a node with minimum time_ready, not
+      // maximum; hence, using a > b for comparison function.
+      return node_state_->at(a).time_ready > node_state_->at(b).time_ready;
+    }
   };
 }