Interface GPUOptions.Experimental.StreamMergeOptionsOrBuilder

All Superinterfaces:
MessageLiteOrBuilder, MessageOrBuilder
All Known Implementing Classes:
GPUOptions.Experimental.StreamMergeOptions, GPUOptions.Experimental.StreamMergeOptions.Builder
Enclosing class:
GPUOptions.Experimental
public static interface GPUOptions.Experimental.StreamMergeOptionsOrBuilder extends MessageOrBuilder

  • Method Details

    • getMergeHostToDeviceStream

      boolean getMergeHostToDeviceStream()
      If true, the compute stream will be used for host_to_device copy as
well. It's no longer necessary to record an event before the copy to
let the copy stream wait for the compute stream to finish. There is
also no need to wait for the copy to complete before executing the
callback function.
      bool merge_host_to_device_stream = 1;
      Returns:
      The mergeHostToDeviceStream.

    • getMergeDeviceToHostStream

      boolean getMergeDeviceToHostStream()
      If true, the compute stream will be used for device_to_host copy as
well. It's no longer necessary to record an event before the copy to
let the copy stream wait for the compute stream to finish.
      bool merge_device_to_host_stream = 2;
      Returns:
      The mergeDeviceToHostStream.

    • getMergeDeviceToDeviceStream

      boolean getMergeDeviceToDeviceStream()
      If true, the compute stream will be used for device_to_device copy as
well. It's no longer necessary to record an event before the copy to
let the copy stream wait for the compute stream of the sending device
to finish. There is also no need to wait for the compute stream of the
receiving device to finish if the copy is within the same device.
      bool merge_device_to_device_stream = 3;
      Returns:
      The mergeDeviceToDeviceStream.