diff options
author | 2018-09-21 19:24:18 -0700 | |
---|---|---|
committer | 2018-09-21 19:31:25 -0700 | |
commit | edbd1a9afc31c9a5127ba769db5e1df11249660a (patch) | |
tree | 90d70735bd2e5018e3dbf13bbbe5f66c9a26a59e /tensorflow/compiler/xla/service/BUILD | |
parent | 0695e9ad8fe6f50942c8c18d648aea982541eeae (diff) |
Add a flag that lets users override the number of host "devices" as seen by XLA
While overriding it from the default of 1 won't be great for performance, it can
help writing tests.
The CL is organized as follows:
* xla/legacy_flags/debug_options_flags and xla/xla.proto now has a
--xla_force_host_platform_device_count flag which defaults to 1.
* xla/service/platform_util.cc respects this --xla_force_host_platform_device_count
flag.
* xla/service/cpu/... has some changes to generalize infeed and outfeed on CPU
to work with multiple devices.
PiperOrigin-RevId: 214078482
Diffstat (limited to 'tensorflow/compiler/xla/service/BUILD')
-rw-r--r-- | tensorflow/compiler/xla/service/BUILD | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/BUILD b/tensorflow/compiler/xla/service/BUILD index 2bc50c70cf..e800cf470c 100644 --- a/tensorflow/compiler/xla/service/BUILD +++ b/tensorflow/compiler/xla/service/BUILD @@ -593,6 +593,7 @@ cc_library( "//tensorflow/compiler/xla:statusor", "//tensorflow/compiler/xla:types", "//tensorflow/compiler/xla:util", + "//tensorflow/compiler/xla/legacy_flags:debug_options_flags", "//tensorflow/core:lib", "//tensorflow/core:stream_executor_no_cuda", "@com_google_absl//absl/strings", |