OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-07 00:20:20 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Add vlogging of HloModule before and after fusion.

Browse Source 
PiperOrigin-RevId: 171029054
This commit is contained in:
Mark Heffernan 2017-10-04 10:38:15 -07:00 committed by TensorFlower Gardener
parent 9e658545a9
commit 4f10a6597c
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/compiler/xla/service/instruction_fusion.cc
View File

@ -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;
}