diff options
| author |  Benoit Steiner <bsteiner@google.com> | 2017-06-13 11:47:33 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-06-13 11:51:31 -0700 |
| commit | 011762e4bd2dc3634a22928b0d75e03a6af612a1 (patch) | |
| tree | 0973f21a3a6fa259e6cc8d2d3f7a90f30b2c950b /tensorflow/core/grappler/costs/virtual_scheduler.cc | |
| parent | 4fe26f75897e99f7f198773008d2736aa05c80b3 (diff) | |
Record the maximum memory usage simulated by the analytical_cost_estimator
PiperOrigin-RevId: 158875735
Diffstat (limited to 'tensorflow/core/grappler/costs/virtual_scheduler.cc')
| -rw-r--r-- | tensorflow/core/grappler/costs/virtual_scheduler.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/costs/virtual_scheduler.cc b/tensorflow/core/grappler/costs/virtual_scheduler.cc index e7f2e5d46b..c68d4e31c4 100644 --- a/tensorflow/core/grappler/costs/virtual_scheduler.cc +++ b/tensorflow/core/grappler/costs/virtual_scheduler.cc @@ -561,6 +561,8 @@ Costs VirtualScheduler::Summary() const { persisent_ops.insert(node->op()); } int64 max_memory_usage = persistent_memory_usage + state.max_memory_usage; + critical_path_costs.estimated_max_memory_per_device[name] = + max_memory_usage; VLOG(1) << "Device = " << name << ", num_nodes = " << state.nodes_executed.size() |