Q/j ddlZGddZy)Nc<eZdZdZd dZdZdZdZdZdZ dZ y ) EventaAWrapper around an MPS event. MPS events are synchronization markers that can be used to monitor the device's progress, to accurately measure timing, and to synchronize MPS streams. Args: enable_timing (bool, optional): indicates if the event should measure time (default: ``False``) cLtjj||_y)N)torch_C_mps_acquireEvent_Event__eventId)self enable_timings R/root/repos/pocs/video-evals/.venv/lib/python3.12/site-packages/torch/mps/event.py__init__zEvent.__init__s33MBcttjdr:|jdkDr*tjj |jyyy)N_mps_releaseEventr)hasattrrrr rr s r __del__z Event.__del__s; 5880 1dnnq6H HH & &t~~ 67I 1rcVtjj|jy)z(Records the event in the default stream.N)rr_mps_recordEventr rs r recordz Event.records !!$..1rcVtjj|jy)zJMakes all future work submitted to the default stream wait for this event.N)rr_mps_waitForEventr rs r waitz Event.waits ""4>>2rcTtjj|jS)zCReturns True if all work currently captured by event has completed.)rr_mps_queryEventr rs r queryz Event.query sxx''77rcVtjj|jy)zWaits until the completion of all work currently captured in this event. This prevents the CPU thread from proceeding until the event completes. N)rr_mps_synchronizeEventr rs r synchronizezEvent.synchronize$s &&t~~6rcjtjj|j|jS)z}Returns the time elapsed in milliseconds after the event was recorded and before the end_event was recorded. )rr_mps_elapsedTimeOfEventsr )r end_events r elapsed_timezEvent.elapsed_time*s%xx00ATATUUrN)F) __name__ __module__ __qualname____doc__r rrrrrr#rr rrs,C7 2387 Vrr)rrr(rr r)s )V)Vr