diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-27 12:52:51 -0700 |
|---|---|---|
| committer |  Gunhan Gulsoy <gunan@google.com> | 2018-06-28 21:37:43 -0700 |
| commit | f5f67296c3430bae595697af3a78460e027cdc6d (patch) | |
| tree | 06c4d7addb7820150d8620e79d41701f5db7b073 /tensorflow/core/protobuf | |
| parent | 15f6b62aeef8292eddd6edbc9ed15cc49774218e (diff) | |
Add GPUOptions::num_dev_to_dev_copy_streams to allow creation of
more than one device-to-device copy stream per GPU device.
This is an experimental feature that will have no effect unless
copy operations explicitly request a stream other than 0, which
currently does not occur anywhere in a standard build.
Eventually it may be of benefit in the presence of multiple
bi-directional concurrent data copies.
PiperOrigin-RevId: 202354513
Diffstat (limited to 'tensorflow/core/protobuf')
| -rw-r--r-- | tensorflow/core/protobuf/config.proto | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/core/protobuf/config.proto b/tensorflow/core/protobuf/config.proto index 7ea422187d..5b6aa47b93 100644 --- a/tensorflow/core/protobuf/config.proto +++ b/tensorflow/core/protobuf/config.proto @@ -143,6 +143,10 @@ message GPUOptions { // multiple processes are sharing a single GPU while individually using less // than 1.0 per process memory fraction. bool use_unified_memory = 2; + + // If > 1, the number of device-to-device copy streams to create + // for each GPUDevice. + int32 num_dev_to_dev_copy_streams = 3; } // Everything inside experimental is subject to change and is not subject |