diff options
author | Mark Heffernan <meheff@google.com> | 2017-10-04 10:38:15 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-10-04 10:46:46 -0700 |
commit | 4f10a6597c12e7274a433ffdef2c00c6891f4c2b (patch) | |
tree | fb87cc4969d0b46963b8d9d35b2c0c252d5e38ab /tensorflow/compiler/xla/service/instruction_fusion.cc | |
parent | 9e658545a91fb8a6cfbcf9cb406d484bcce4586f (diff) |
Add vlogging of HloModule before and after fusion.
PiperOrigin-RevId: 171029054
Diffstat (limited to 'tensorflow/compiler/xla/service/instruction_fusion.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/instruction_fusion.cc | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/instruction_fusion.cc b/tensorflow/compiler/xla/service/instruction_fusion.cc index 177d2e2a93..7a27381642 100644 --- a/tensorflow/compiler/xla/service/instruction_fusion.cc +++ b/tensorflow/compiler/xla/service/instruction_fusion.cc @@ -203,6 +203,9 @@ bool InstructionFusion::CanFuseOnAllPaths( } StatusOr<bool> InstructionFusion::Run(HloModule* module) { + VLOG(2) << "Before instruction fusion:"; + XLA_VLOG_LINES(2, module->ToString()); + bool changed = false; module_ = module; for (auto* computation : module->MakeNonfusionComputations()) { @@ -371,6 +374,10 @@ StatusOr<bool> InstructionFusion::Run(HloModule* module) { } } } + + VLOG(2) << "After instruction fusion:"; + XLA_VLOG_LINES(2, module->ToString()); + return changed; } |