diff options
author | 2018-07-09 01:49:04 -0700 | |
---|---|---|
committer | 2018-07-09 01:54:33 -0700 | |
commit | 955e356e4c69d3fce4ac2bac5966671e964f9627 (patch) | |
tree | 95183ec6d5122e5dae704b8a80796f01dfa1b85a /tensorflow/compiler/xla/service/hlo_instructions.cc | |
parent | caf711b6be448d46354e0cbef23989fa837efcf7 (diff) |
[SE,XLA] Switch to using multiple streams in xla_device_context
Instead of having one stream for compute, host-to-device and device-to-host transfers, switch to having separate streams, just like the GPU does.
Add a se::Event field to XlaTensor to allow accurate inter-stream dependencies to be created.
As part of this:
- Fix TransferManager::TransferLiteralFrom/ToDevice to correctly make generated substreams wait on their master stream.
- Fix Stream::BlockHostUntilDone() to not block on or return substreams. This behavior is completely broken and not only nondeterministically returns substreams to the pool but causes indefinite hangs with the HostStream.
PiperOrigin-RevId: 203726543
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_instructions.cc')
0 files changed, 0 insertions, 0 deletions