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.