diff options
author | 2018-01-19 15:01:35 -0800 | |
---|---|---|
committer | 2018-01-19 15:05:11 -0800 | |
commit | a141651809e17be5ebe02715074357a7c52c0220 (patch) | |
tree | fd0046b2b5b289fb8162ce917f40a7ef3d064fda /tensorflow/compiler/xla/legacy_flags | |
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/legacy_flags')
-rw-r--r-- | tensorflow/compiler/xla/legacy_flags/debug_options_flags.cc | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/legacy_flags/debug_options_flags.cc b/tensorflow/compiler/xla/legacy_flags/debug_options_flags.cc index e88bffd0ba..fe3a4d2f6d 100644 --- a/tensorflow/compiler/xla/legacy_flags/debug_options_flags.cc +++ b/tensorflow/compiler/xla/legacy_flags/debug_options_flags.cc @@ -224,6 +224,11 @@ void AllocateFlags() { "xla_dump_hlo_proto_to", flag_values->mutable_xla_dump_hlo_proto_to(), "Dump compilation artifacts as proto binary into this directory."), tensorflow::Flag( + "xla_dump_prepass_hlo_proto_to", + flag_values->mutable_xla_dump_prepass_hlo_proto_to(), + "Dump compilation artifacts, before hlo passes are executed, as " + "proto binary into this directory."), + tensorflow::Flag( "xla_test_all_output_layouts", bool_setter_for(&DebugOptions::set_xla_test_all_output_layouts), flag_values->xla_test_all_output_layouts(), |