a ¡º”hÂã@sddlZGdd„dƒZdS)éNc@sxeZdZdZdeddœdd„Zddœdd „Zddœd d „Zddœd d „Zedœdd„Z ddœdd„Z de dœdd„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``) FN)Ú enable_timingÚreturncCstj |¡|_dS)N)ÚtorchÚ_CZ_mps_acquireEventÚ_Event__eventId)Úselfr©r ú=/var/www/auris/lib/python3.9/site-packages/torch/mps/event.pyÚ__init__szEvent.__init__)rcCs(ttjdƒr$|jdkr$tj |j¡dS)NÚ_mps_releaseEventr)Úhasattrrrrr ©rr r r Ú__del__sz Event.__del__cCstj |j¡dS)z(Records the event in the default stream.N)rrZ_mps_recordEventrrr r r Úrecordsz Event.recordcCstj |j¡dS)zJMakes all future work submitted to the default stream wait for this event.N)rrZ_mps_waitForEventrrr r r Úwaitsz Event.waitcCstj |j¡S)zCReturns True if all work currently captured by event has completed.)rrZ_mps_queryEventrrr r r Úquerysz Event.querycCstj |j¡dS)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#szEvent.synchronize)Ú end_eventrcCstj |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)szEvent.elapsed_time)F) Ú__name__Ú __module__Ú __qualname__Ú__doc__Úboolr rrrrrÚfloatrr r r r rs r)rrr r r r Ús