diff options
author | 2018-01-24 12:12:02 -0800 | |
---|---|---|
committer | 2018-01-24 12:15:50 -0800 | |
commit | 02c2214b8da916a4a9e9eb3fde2711e49e7b4703 (patch) | |
tree | 965383e084bec7465fe7bc2421466c9be0b51300 /tensorflow/compiler/xla/service/hlo_ordering.cc | |
parent | 94faa1e921d647f321ca1b3204a25aa0dbede856 (diff) |
[XLA] Fix crash in HloOrdering::ToString().
HloOrdering::ToString() was crashing when given a computation containing
a fusion instruction.
PiperOrigin-RevId: 183121645
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_ordering.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/hlo_ordering.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_ordering.cc b/tensorflow/compiler/xla/service/hlo_ordering.cc index 6f6e679a21..68e3c9618c 100644 --- a/tensorflow/compiler/xla/service/hlo_ordering.cc +++ b/tensorflow/compiler/xla/service/hlo_ordering.cc @@ -249,7 +249,7 @@ bool PredecessorHloOrdering::ExecutesBeforeInSameComputation( string PredecessorHloOrdering::ToStringHelper(const string& name) const { std::vector<string> pieces; pieces.push_back(name); - for (auto* computation : module_->computations()) { + for (auto* computation : module_->MakeNonfusionComputations()) { pieces.push_back(tensorflow::strings::Printf("computation %s:", computation->name().c_str())); const auto all = computation->MakeInstructionPostOrder(); |