
    khI	                     <   U d dl mZ d dlmZ  ed      Zded<    ed      Zded<    ed      Zd	ed
<    ed      Zd	ed<    ed      Z	ded<    ed      Z
ded<    ed      Zded<    ed      Zded<    ed      Zded<    ed      Zded<   deegdf   ddfdZdeegdf   ddfdZdeeegdf   ddfdZdeeegdf   ddfd Zdeegdf   ddfd!Zdeegdf   ddfd"Zdeegdf   ddfd#Zdeg df   ddfd$Zdeegdf   ddfd%Zdeegdf   ddfd&Zy)'    )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                 .    t         j                  |        y N)r   add_callbackr   s    Q/var/www/teggl/fontify/venv/lib/python3.12/site-packages/torch/cuda/_gpu_trace.py$register_callback_for_event_creationr   $       ''+    c                 .    t         j                  |        y r   )r   r   r   s    r   $register_callback_for_event_deletionr   (   r   r   c                 .    t         j                  |        y r   )r   r   r   s    r   "register_callback_for_event_recordr   ,   s    %%b)r   c                 .    t         j                  |        y r   )r   r   r   s    r    register_callback_for_event_waitr   0   s    ##B'r   c                 .    t         j                  |        y r   )r	   r   r   s    r   'register_callback_for_memory_allocationr    4   s    **2.r   c                 .    t         j                  |        y r   )r
   r   r   s    r   )register_callback_for_memory_deallocationr"   8   s    ,,R0r   c                 .    t         j                  |        y r   )r   r   r   s    r   %register_callback_for_stream_creationr$   <   s    ((,r   c                 .    t         j                  |        y r   )r   r   r   s    r   ,register_callback_for_device_synchronizationr&   @       "//3r   c                 .    t         j                  |        y r   )r   r   r   s    r   ,register_callback_for_stream_synchronizationr)   D   r'   r   c                 .    t         j                  |        y 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   