Clear reachability map after fusion.

PiperOrigin-RevId: 331830813 Change-Id: I9b52a37ab321977ae65244b24ad4945238247cb8
2020-09-15 12:39:51 -07:00 · 2020-09-15 12:39:51 -07:00 · 9f2dbad68e
commit 9f2dbad68e
parent 9f96186327
1 changed files with 3 additions and 0 deletions
--- a/tensorflow/compiler/xla/service/instruction_fusion.cc
+++ b/tensorflow/compiler/xla/service/instruction_fusion.cc
@ -602,6 +602,9 @@ StatusOr<bool> InstructionFusion::Run(HloModule* module) {
    VLOG(1) << FusionConfigToString(*fusion_config);
    module->set_config(module_config);
  }
+
+  reachability_.reset();
+
  VLOG(1) << "Fusion count: " << fuse_count;

  return changed;