diff options
author | Sanjoy Das <sanjoy@google.com> | 2018-09-21 19:24:18 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-21 19:31:25 -0700 |
commit | edbd1a9afc31c9a5127ba769db5e1df11249660a (patch) | |
tree | 90d70735bd2e5018e3dbf13bbbe5f66c9a26a59e /CODE_OF_CONDUCT.md | |
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 'CODE_OF_CONDUCT.md')
0 files changed, 0 insertions, 0 deletions