diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-08-23 15:07:04 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-23 15:12:27 -0700 |
commit | fb7b8458751cc0ef38e4c98430517ab134f85aca (patch) | |
tree | b1673e7f6cb0e6d4d91ca660f9f7cc6fe289dd36 /tensorflow/core/grappler/costs/virtual_scheduler.cc | |
parent | 91aa51fbd2c55a45193c309b34c46f35d48b0167 (diff) |
Sorted the per-device summary printout with device names to improve readability.
PiperOrigin-RevId: 210007888
Diffstat (limited to 'tensorflow/core/grappler/costs/virtual_scheduler.cc')
-rw-r--r-- | tensorflow/core/grappler/costs/virtual_scheduler.cc | 11 |
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 6e3ebdee12..037a823096 100644 --- a/tensorflow/core/grappler/costs/virtual_scheduler.cc +++ b/tensorflow/core/grappler/costs/virtual_scheduler.cc @@ -880,10 +880,15 @@ Costs VirtualScheduler::Summary() const { // Print per device summary VLOG(1) << "Devices:"; Costs critical_path_costs = Costs::ZeroCosts(); + std::vector<string> device_names; + device_names.reserve(device_.size()); + for (auto& it : device_) { + device_names.push_back(it.first); + } + std::sort(device_names.begin(), device_names.end()); - for (const auto& device : device_) { - const auto& name = device.first; - const auto& state = device.second; + for (const auto& name : device_names) { + const auto& state = device_.at(name); std::map<string, int64> op_to_memory; // First profile only persistent memory usage. |