diff options
Diffstat (limited to 'tensorflow/compiler/xla/service/buffer_liveness.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/buffer_liveness.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/buffer_liveness.cc b/tensorflow/compiler/xla/service/buffer_liveness.cc index 6720a90ef8..f085ffa6bc 100644 --- a/tensorflow/compiler/xla/service/buffer_liveness.cc +++ b/tensorflow/compiler/xla/service/buffer_liveness.cc @@ -47,6 +47,9 @@ StatusOr<std::unique_ptr<BufferLiveness>> BufferLiveness::Run( tensorflow::Status BufferLiveness::Analyze() { TF_ASSIGN_OR_RETURN(points_to_analysis_, TuplePointsToAnalysis::Run(module_)); for (auto& computation : module_->computations()) { + if (computation->IsFusionComputation()) { + continue; + } // Gather all instructions whose buffers might alias other instructions into // the set aliased_buffers_. This includes those contained as a tuple // element in other instruction's output. |