aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/compiler/xla/service/hlo_computation.cc
diff options
context:
space:
mode:
authorGravatar Chris Leary <leary@google.com>2017-09-29 12:13:44 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-09-29 12:17:48 -0700
commit76db7553ab2998116a62d6c242aa39373a362993 (patch)
tree7da6a5dd9d2bd879157c957e1d53cece03195071 /tensorflow/compiler/xla/service/hlo_computation.cc
parent2c2068e795cf5129062cf61786b8d5e89ae7a7b3 (diff)
[XLA] Make it possible to inline calls to side-effecting computations.
PiperOrigin-RevId: 170515496
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_computation.cc')
-rw-r--r--tensorflow/compiler/xla/service/hlo_computation.cc3
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_computation.cc b/tensorflow/compiler/xla/service/hlo_computation.cc
index 3e2a8d9264..444104d88f 100644
--- a/tensorflow/compiler/xla/service/hlo_computation.cc
+++ b/tensorflow/compiler/xla/service/hlo_computation.cc
@@ -198,7 +198,8 @@ Status HloComputation::RemoveInstructionAndUnusedOperands(
TF_RET_CHECK(root_instruction() != instruction);
TF_RET_CHECK(instruction->user_count() == 0);
- TF_RET_CHECK(IsRemovable(instruction));
+ TF_RET_CHECK(IsRemovable(instruction))
+ << "Cannot remove instruction: " << instruction->ToString();
std::unordered_set<HloInstruction*> removed;
std::queue<HloInstruction*> worklist;
worklist.push(instruction);