o i@sddlZGdddZdS)Nc@sleZdZdZddeddfddZddd Zdd d Zdd d ZdefddZ dddZ ddde fddZ dS)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``) F enable_timingreturnNcCstj||_dS)N)torch_C_mps_acquireEvent_Event__eventId)selfrr L/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/torch/mps/event.py__init__szEvent.__init__cCs0ttjdr|jdkrtj|jdSdSdS)N_mps_releaseEventr)hasattrrrrr r r r r __del__sz Event.__del__cCtj|jdS)z(Records the event in the default stream.N)rr_mps_recordEventrrr r r recordz Event.recordcCr)zJMakes all future work submitted to the default stream wait for this event.N)rr_mps_waitForEventrrr r r waitrz Event.waitcCstj|jS)zCReturns True if all work currently captured by event has completed.)rr_mps_queryEventrrr r r querysz Event.querycCr)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_synchronizeEventrrr r r synchronize#zEvent.synchronize end_eventcCstj|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 rr r r elapsed_time)rzEvent.elapsed_time)F)rN) __name__ __module__ __qualname____doc__boolr rrrrrfloatrr r r r rs    r)rrr r r r s