
    ,hI	                     P   % S SK Jr  S SKJr  \" S5      rS\S'   \" S5      rS\S'   \" S5      rS	\S
'   \" S5      rS	\S'   \" S5      r	S\S'   \" S5      r
S\S'   \" S5      rS\S'   \" S5      rS\S'   \" S5      rS\S'   \" S5      rS\S'   S\\/S4   SS4S jrS\\/S4   SS4S jrS\\\/S4   SS4S jrS\\\/S4   SS4S  jrS\\/S4   SS4S! jrS\\/S4   SS4S" jrS\\/S4   SS4S# jrS\/ S4   SS4S$ jrS\\/S4   SS4S% jrS\\/S4   SS4S& jrg)'    )Callable)CallbackRegistryzCUDA event creationzCallbackRegistry[int]EventCreationCallbackszCUDA event deletionEventDeletionCallbackszCUDA event recordzCallbackRegistry[int, int]EventRecordCallbackszCUDA event waitEventWaitCallbackszCUDA memory allocationMemoryAllocationCallbackszCUDA memory deallocationMemoryDeallocationCallbackszCUDA stream creationStreamCreationCallbackszCUDA device synchronizationzCallbackRegistry[[]]DeviceSynchronizationCallbackszCUDA stream synchronizationStreamSynchronizationCallbackszCUDA event synchronizationEventSynchronizationCallbackscbNreturnc                 .    [         R                  U 5        g N)r   add_callbackr   s    M/var/www/html/shao/venv/lib/python3.13/site-packages/torch/cuda/_gpu_trace.py$register_callback_for_event_creationr   $       ''+    c                 .    [         R                  U 5        g r   )r   r   r   s    r   $register_callback_for_event_deletionr   (   r   r   c                 .    [         R                  U 5        g r   )r   r   r   s    r   "register_callback_for_event_recordr   ,   s    %%b)r   c                 .    [         R                  U 5        g r   )r   r   r   s    r    register_callback_for_event_waitr   0   s    ##B'r   c                 .    [         R                  U 5        g r   )r	   r   r   s    r   'register_callback_for_memory_allocationr    4   s    **2.r   c                 .    [         R                  U 5        g r   )r
   r   r   s    r   )register_callback_for_memory_deallocationr"   8   s    ,,R0r   c                 .    [         R                  U 5        g r   )r   r   r   s    r   %register_callback_for_stream_creationr$   <   s    ((,r   c                 .    [         R                  U 5        g r   )r   r   r   s    r   ,register_callback_for_device_synchronizationr&   @       "//3r   c                 .    [         R                  U 5        g r   )r   r   r   s    r   ,register_callback_for_stream_synchronizationr)   D   r'   r   c                 .    [         R                  U 5        g r   )r   r   r   s    r   +register_callback_for_event_synchronizationr+   H   s    !..r2r   )typingr   torch._utilsr   r   __annotations__r   r   r   r	   r
   r   r   r   r   intr   r   r   r   r    r"   r$   r&   r)   r+    r   r   <module>r1      s.    ) 3C3 /  3C3 /  6F6 2  4DDU3V 0 V5E6 2  8H8 4  4D4 0  :J!:  6  ;K!;  7  :J : 6 
,XseTk-B ,t ,,XseTk-B ,t ,*8S#J4D+E *$ *(3*d2B)C ( (/#0E /$ /1(C5$;2G 1D 1-hud{.C - -4Xb$h5G 4D 44XseTk5J 4t 43HcUD[4I 3d 3r   