diff options
| author |  Eli Bendersky <eliben@google.com> | 2017-06-16 12:37:05 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-06-16 12:40:16 -0700 |
| commit | 12efd3d0bbea953e52aee12eb5a3d5d2269ec16a (patch) | |
| tree | a129c2898e04d437d716f7a95ed545bc5db07e5b /tensorflow/compiler/xla/service/local_service.cc | |
| parent | 1f7990e6d762a0444b1c0a94c7e70e7cf5c9d0a8 (diff) | |
[XLA] Remove gpu/gpu_backend-specific flags
Move useful flags into debug_options, and leave some less used flags out - they
can be propagated through debug_options if required (for now there's too much
duplication between them and what's already inside)
PiperOrigin-RevId: 159261661
Diffstat (limited to 'tensorflow/compiler/xla/service/local_service.cc')
| -rw-r--r-- | tensorflow/compiler/xla/service/local_service.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/local_service.cc b/tensorflow/compiler/xla/service/local_service.cc index 748124ebc7..ffc775e564 100644 --- a/tensorflow/compiler/xla/service/local_service.cc +++ b/tensorflow/compiler/xla/service/local_service.cc @@ -19,6 +19,7 @@ limitations under the License. #include <utility> #include <vector> +#include "tensorflow/compiler/xla/legacy_flags/debug_options_flags.h" #include "tensorflow/compiler/xla/legacy_flags/service_flags.h" #include "tensorflow/compiler/xla/ptr_util.h" #include "tensorflow/compiler/xla/service/backend.h" @@ -153,6 +154,7 @@ StatusOr<std::unique_ptr<Executable>> LocalService::CompileExecutable( auto module_config = MakeUnique<HloModuleConfig>(*program_shape); module_config->set_has_hybrid_result(has_hybrid_result); module_config->set_replica_count(execute_backend_->replica_count()); + module_config->set_debug_options(legacy_flags::GetDebugOptionsFromFlags()); legacy_flags::ServiceFlags* flags = legacy_flags::GetServiceFlags(); if (flags->xla_hlo_profile) { module_config->enable_hlo_profiling(true); |