Yh(ddlZGddZdS)Nc`eZdZdZddeddfdZddZddZdd Zdefd Z dd Z d dde fd 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_timingreturnNcNtj||_dS)N)torch_C_mps_acquireEvent_Event__eventId)selfrs a/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/site-packages/torch/mps/event.py__init__zEvent.__init__s33MBBcttjdr1|jdkr(tj|jdSdSdS)N_mps_releaseEventr)hasattrrr r rr s r __del__z Event.__del__sQ 580 1 1 7dnq6H6H H & &t~ 6 6 6 6 6 7 76H6HrcNtj|jdS)z(Records the event in the default stream.N)rr _mps_recordEventr rs r recordz Event.records  !!$.11111rcNtj|jdS)zJMakes all future work submitted to the default stream wait for this event.N)rr _mps_waitForEventr rs r waitz Event.waits  ""4>22222rcJtj|jS)zCReturns True if all work currently captured by event has completed.)rr _mps_queryEventr rs r queryz Event.querysx''777rcNtj|jdS)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~66666r end_eventcVtj|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 r!s r elapsed_timezEvent.elapsed_time)s!x00ATUUUr)F)rN) __name__ __module__ __qualname____doc__boolrrrrrr floatr$rr rrsCCdCtCCCC7777 222233338t88887777 VgV%VVVVVVrr)rrr+rr r,sJ )V)V)V)V)V)V)V)V)V)Vr