Event#

classtorch.mps.event.Event(enable_timing=False)[source]#

Wrapper around an MPS event.

MPS events are synchronization markers that can be used to monitor thedevice’s progress, to accurately measure timing, and to synchronize MPS streams.

Parameters

enable_timing (bool,optional) – indicates if the event should measure time(default:False)

elapsed_time(end_event)[source]#

Returns the time elapsed in milliseconds after the event wasrecorded and before the end_event was recorded.

Return type

float

query()[source]#

Returns True if all work currently captured by event has completed.

Return type

bool

record()[source]#

Records the event in the default stream.

synchronize()[source]#

Waits until the completion of all work currently captured in this event.This prevents the CPU thread from proceeding until the event completes.

wait()[source]#

Makes all future work submitted to the default stream wait for this event.