`iI UddlmZddlmZedZded<edZded<edZd ed <ed Zd ed <ed Z ded<edZ ded<edZ ded<edZ ded<edZ ded<edZded<deegdfddfdZdeegdfddfdZdeeegdfddfdZdeeegdfddfd Zdeegdfddfd!Zdeegdfddfd"Zdeegdfddfd#Zdegdfddfd$Zdeegdfddfd%Zdeegdfddfd&ZdS)')Callable)CallbackRegistryzCUDA event creationzCallbackRegistry[int]EventCreationCallbackszCUDA event deletionEventDeletionCallbackszCUDA event recordzCallbackRegistry[int, int]EventRecordCallbackszCUDA event waitEventWaitCallbackszCUDA memory allocationMemoryAllocationCallbackszCUDA memory deallocationMemoryDeallocationCallbackszCUDA stream creationStreamCreationCallbackszCUDA device synchronizationzCallbackRegistry[[]]DeviceSynchronizationCallbackszCUDA stream synchronizationStreamSynchronizationCallbackszCUDA event synchronizationEventSynchronizationCallbackscbNreturnc:t|dSN)r add_callbackrs i/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torch/cuda/_gpu_trace.py$register_callback_for_event_creationr$''+++++c:t|dSr)rrrs r$register_callback_for_event_deletionr(rrc:t|dSr)rrrs r"register_callback_for_event_recordr,s%%b)))))rc:t|dSr)rrrs r register_callback_for_event_waitr0s##B'''''rc:t|dSr)r rrs r'register_callback_for_memory_allocationr 4s**2.....rc:t|dSr)r rrs r)register_callback_for_memory_deallocationr"8s,,R00000rc:t|dSr)r rrs r%register_callback_for_stream_creationr$<s((,,,,,rc:t|dSr)r rrs r,register_callback_for_device_synchronizationr&@"//33333rc:t|dSr)r rrs r,register_callback_for_stream_synchronizationr)Dr'rc:t|dSr)rrrs r+register_callback_for_event_synchronizationr+Hs!..r22222r)typingr torch._utilsrr__annotations__rrrr r r r r rintrrrrr r"r$r&r)r+rrr1s))))))3C2B33/3C2B33/6F5E6624D3CDU3V3V0VVV5E5E6628H7G8844D3C440:J9I!:: 6;K:J!;; 7:J9I ::6 ,XseTk-B,t,,,,,XseTk-B,t,,,,*8S#J4D+E*$****(3*d2B)C(((((/#0E/$////1(C5$;2G1D1111-hud{.C-----4Xb$h5G4D44444XseTk5J4t44443HcUD[4I3d333333r