o wZŽhÂã@sddlZGdd„dƒZdS)éNc@sleZdZdZddeddfdd„Zddd „Zdd d „Zdd d „Zdefdd„Z ddd„Z ddde fdd„Z 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_timingÚreturnNcCstj |¡|_dS)N)ÚtorchÚ_CZ_mps_acquireEventÚ_Event__eventId)Úselfr©r ú>/var/www/auris/lib/python3.10/site-packages/torch/mps/event.pyÚ__init__szEvent.__init__cCs0ttjdƒr|jdkrtj |j¡dSdSdS)NÚ_mps_releaseEventr)Úhasattrrrrr ©rr r r Ú__del__sÿz Event.__del__cCótj |j¡dS)z(Records the event in the default stream.N)rrZ_mps_recordEventrrr r r Úrecordóz Event.recordcCr)zJMakes all future work submitted to the default stream wait for this event.N)rrZ_mps_waitForEventrrr r r Úwaitrz Event.waitcCstj |j¡S)zCReturns True if all work currently captured by event has completed.)rrZ_mps_queryEventrrr r r Úquerysz Event.querycCr)z¡Waits until the completion of all work currently captured in this event. This prevents the CPU thread from proceeding until the event completes. N)rrZ_mps_synchronizeEventrrr r r Ú synchronize#ózEvent.synchronizeÚ end_eventcCstj |j|j¡S)z}Returns the time elapsed in milliseconds after the event was recorded and before the end_event was recorded. )rrZ_mps_elapsedTimeOfEventsr)rrr r r Ú elapsed_time)rzEvent.elapsed_time)F)rN) Ú__name__Ú __module__Ú __qualname__Ú__doc__Úboolr rrrrrÚfloatrr r r r rs    r)rrr r r r Ús