experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[XLA] Correct WhileLoopInvariantCodeMotion log messages.

Browse Source 
PiperOrigin-RevId: 310425877
Change-Id: I582ca2b53515d932c7bde72d4ec91986b1482c43
This commit is contained in:
A. Unique TensorFlower 2020-05-07 13:28:41 -07:00 committed by TensorFlower Gardener
parent 2acaff3d89
commit 8fef4ea574
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/compiler/xla/service/while_loop_invariant_code_motion.cc
View File

@ -300,7 +300,7 @@ WhileLoopInvariantCodeMotion::TryHoistingInvariantInstructionsFromWhileBody(
} }
StatusOr<bool> WhileLoopInvariantCodeMotion::Run(HloModule* module) { StatusOr<bool> WhileLoopInvariantCodeMotion::Run(HloModule* module) {
VLOG(2) << "HLO module before WhileLoopConstantSinking:"; VLOG(2) << "HLO module before WhileLoopInvariantCodeMotion:";
XLA_VLOG_LINES(2, module->ToString()); XLA_VLOG_LINES(2, module->ToString());
bool changed = false; bool changed = false;
@ -332,10 +332,10 @@ StatusOr<bool> WhileLoopInvariantCodeMotion::Run(HloModule* module) {
} }
if (changed) { if (changed) {
VLOG(2) << "HLO module after WhileLoopConstantSinking:"; VLOG(2) << "HLO module after WhileLoopInvariantCodeMotion:";
XLA_VLOG_LINES(2, module->ToString()); XLA_VLOG_LINES(2, module->ToString());
} else { } else {
VLOG(2) << "HLO module unchanged after WhileLoopConstantSinking"; VLOG(2) << "HLO module unchanged after WhileLoopInvariantCodeMotion";
} }
return changed; return changed;