diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-07-18 19:15:53 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-07-18 19:19:31 -0700 |
| commit | 9293c557bd2df05658727418067ccee7a77a4be3 (patch) | |
| tree | 1cdd3d8da30e787b5e476543cebce144e44eb89c /tensorflow/compiler/xla/service/local_service.cc | |
| parent | 2aaa620f8cfcf9df22d4be3bd85954a217faa818 (diff) | |
[XLA] Get rid of ServiceFlags by absorbing it into DebugOptions.
After this change HloModuleConfig::hlo_profiling_enabled_ is redundant. I'll
remove it in a future change.
PiperOrigin-RevId: 162436163
Diffstat (limited to 'tensorflow/compiler/xla/service/local_service.cc')
| -rw-r--r-- | tensorflow/compiler/xla/service/local_service.cc | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/tensorflow/compiler/xla/service/local_service.cc b/tensorflow/compiler/xla/service/local_service.cc index 25588a6fb8..45e37c6f65 100644 --- a/tensorflow/compiler/xla/service/local_service.cc +++ b/tensorflow/compiler/xla/service/local_service.cc @@ -20,7 +20,6 @@ limitations under the License. #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" #include "tensorflow/compiler/xla/service/computation_layout.h" @@ -151,8 +150,7 @@ StatusOr<std::unique_ptr<Executable>> LocalService::CompileExecutable( module_config->set_intra_op_parallelism_threads( execute_backend_->eigen_intra_op_thread_pool()->NumThreads()); } - legacy_flags::ServiceFlags* flags = legacy_flags::GetServiceFlags(); - if (flags->xla_hlo_profile) { + if (module_config->debug_options().xla_hlo_profile()) { module_config->enable_hlo_profiling(true); } auto* computation_layout = module_config->mutable_entry_computation_layout(); |