diff options
author | Roy Frostig <frostig@google.com> | 2018-06-01 14:56:17 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-01 14:58:58 -0700 |
commit | 441979ff0399418b7883ca6c267c08fc716ce74b (patch) | |
tree | dc0cf6ddf7467377ecb4ebc140b6c1f5bd9ebd42 /tensorflow/compiler/xla/service/local_service.cc | |
parent | d1a3c24745aaf54098b7de3069d65fa92002b221 (diff) |
[XLA] Add an unoptimized HLO output flag to ExecutableBuildOptions and to the XLA local Python client.
PiperOrigin-RevId: 198930874
Diffstat (limited to 'tensorflow/compiler/xla/service/local_service.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/local_service.cc | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/local_service.cc b/tensorflow/compiler/xla/service/local_service.cc index 375c4a6780..1d9c9e0678 100644 --- a/tensorflow/compiler/xla/service/local_service.cc +++ b/tensorflow/compiler/xla/service/local_service.cc @@ -108,6 +108,11 @@ ExecutionOptions CreateExecutionOptions( ->set_xla_dump_optimized_hlo_proto_to( build_options.dump_optimized_hlo_proto_to().value()); } + if (build_options.dump_unoptimized_hlo_proto_to().has_value()) { + execution_options.mutable_debug_options() + ->set_xla_dump_unoptimized_hlo_proto_to( + build_options.dump_unoptimized_hlo_proto_to().value()); + } if (build_options.dump_per_pass_hlo_proto_to().has_value()) { execution_options.mutable_debug_options() ->set_xla_dump_per_pass_hlo_proto_to( |