diff options
Diffstat (limited to 'tensorflow/compiler/xla/service/algebraic_simplifier.cc')
| -rw-r--r-- | tensorflow/compiler/xla/service/algebraic_simplifier.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/algebraic_simplifier.cc b/tensorflow/compiler/xla/service/algebraic_simplifier.cc index 4837402c15..691f9f2296 100644 --- a/tensorflow/compiler/xla/service/algebraic_simplifier.cc +++ b/tensorflow/compiler/xla/service/algebraic_simplifier.cc @@ -1586,6 +1586,9 @@ StatusOr<bool> AlgebraicSimplifier::Run(HloModule* module) { // module, invalidating iteration. std::vector<HloComputation*> computations; for (auto& comp : module->computations()) { + if (comp->IsFusionComputation()) { + continue; + } computations.push_back(comp.get()); } for (auto& comp : computations) { |