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

Log Hlo IR during AOT compilation

Browse Source 
PiperOrigin-RevId: 172944165
This commit is contained in:
Sanjoy Das 2017-10-20 15:40:34 -07:00 committed by TensorFlower Gardener
parent 985031a101
commit ccfd9c1e50
1 changed files with 6 additions and 0 deletions

6
tensorflow/compiler/xla/service/cpu/cpu_compiler.cc
View File

@ -757,8 +757,14 @@ CpuCompiler::CompileAheadOfTime(std::vector<std::unique_ptr<HloModule>> modules,
HloModule* module = modules[i].get(); HloModule* module = modules[i].get();
VLOG(1) << "Compiling ahead-of-time: " << module->name(); VLOG(1) << "Compiling ahead-of-time: " << module->name();
VLOG(2) << "Before optimization:";
XLA_VLOG_LINES(2, module->ToString());
TF_RETURN_IF_ERROR(RunHloPasses(module, /*is_aot_compile=*/true)); TF_RETURN_IF_ERROR(RunHloPasses(module, /*is_aot_compile=*/true));
VLOG(2) << "After optimization:";
XLA_VLOG_LINES(2, module->ToString());
TF_ASSIGN_OR_RETURN( TF_ASSIGN_OR_RETURN(
SequentialHloOrdering::HloModuleSequence module_sequence, SequentialHloOrdering::HloModuleSequence module_sequence,
CreateMemoryMinimizingSequence(*module, BufferSizeBytesFunction())); CreateMemoryMinimizingSequence(*module, BufferSizeBytesFunction()));