synchronize¶

dragon.cuda.synchronize(
  device_index=None,
  stream_index=0
)[source]¶

Synchronize a specified CUDA stream.

If device_index is None, the current device will be selected.

Parameters:

• device_index (int, optional) – The device index.
• stream_index (int, optional, default=0) – The stream index.