diff options
| author |  Jeremy Lau <lauj@google.com> | 2018-05-16 15:54:49 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-16 15:59:29 -0700 |
| commit | 250415665dbd6ea200e8fc17e1c61eaf32312343 (patch) | |
| tree | a1655b82ae0d54808c3c9a21e8838a1cbe602cc5 /tensorflow/compiler/xla/service/heap_simulator.cc | |
| parent | 9fd3485db92d6bfee928dfaaba3dc69938bab8b6 (diff) | |
Move DoesNotUseOperandBuffer and CanShareOperandBufferWithUser from
liveness_util to methods on TuplePointsToAnalysis and HloDataflowAnalysis.
PiperOrigin-RevId: 196903216
Diffstat (limited to 'tensorflow/compiler/xla/service/heap_simulator.cc')
| -rw-r--r-- | tensorflow/compiler/xla/service/heap_simulator.cc | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/tensorflow/compiler/xla/service/heap_simulator.cc b/tensorflow/compiler/xla/service/heap_simulator.cc index 9a07ee3683..06a5e0351b 100644 --- a/tensorflow/compiler/xla/service/heap_simulator.cc +++ b/tensorflow/compiler/xla/service/heap_simulator.cc @@ -19,7 +19,6 @@ limitations under the License. #include <vector> #include "tensorflow/compiler/xla/map_util.h" -#include "tensorflow/compiler/xla/service/liveness_util.h" #include "tensorflow/compiler/xla/util.h" namespace xla { @@ -203,9 +202,9 @@ Status HeapSimulator::RunComputation( for (const BufferValue* operand_buffer : operand_buffers_to_free) { if (buffer->instruction()->IsUserOf(operand_buffer->instruction()) && buffer->instruction()->opcode() != HloOpcode::kCopy && - CanShareOperandBufferWithUser( + points_to_analysis.CanShareOperandBufferWithUser( operand_buffer->instruction(), operand_buffer->index(), - buffer->instruction(), buffer->index(), points_to_analysis)) { + buffer->instruction(), buffer->index())) { VLOG(3) << " Sharing: " << buffer->ToString() << " with " << operand_buffer->ToString(); ShareBuffer(buffer, operand_buffer, instruction); |