synchronize

dragon.cuda.synchronize(
  device_id=None,
  stream_id=0
)[source]

Synchronize the specified stream.

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

Parameters:
  • device_id (int, optional) – The device index.
  • stream_id (int, optional, default=0) – The stream index.