diff options
| author |  Nick Desaulniers <ndesaulniers@google.com> | 2018-01-19 15:01:35 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-19 15:05:11 -0800 |
| commit | a141651809e17be5ebe02715074357a7c52c0220 (patch) | |
| tree | fd0046b2b5b289fb8162ce917f40a7ef3d064fda /tensorflow/compiler/xla/service/compile_only_service.cc | |
| parent | 0c4c353690b0cec2e26e10abd59f66f7e61cc974 (diff) | |
[XLA] Add xla_dump_prepass_hlo_proto_to debug flag
Add xla_dump_prepass_hlo_proto_to debug flag for dumping HLO protos *BEFORE*
running HLO passes, which will perform backend specific optimizations.
PiperOrigin-RevId: 182594324
Diffstat (limited to 'tensorflow/compiler/xla/service/compile_only_service.cc')
| -rw-r--r-- | tensorflow/compiler/xla/service/compile_only_service.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/compile_only_service.cc b/tensorflow/compiler/xla/service/compile_only_service.cc index 9e96898d9b..b9306a8bb0 100644 --- a/tensorflow/compiler/xla/service/compile_only_service.cc +++ b/tensorflow/compiler/xla/service/compile_only_service.cc @@ -107,6 +107,7 @@ CompileOnlyService::CompileAheadOfTime( computation_tracker_.BuildHloModule( versioned_handle, *module_config, /*include_unreachable_instructions=*/true)); + TF_RETURN_IF_ERROR(MaybeDumpHloModule(*hlo_module)); hlo_modules.push_back(std::move(hlo_module)); } |