kh3 <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&Zy)')Callable)CallbackRegistryzXPU event creationzCallbackRegistry[int]EventCreationCallbackszXPU event deletionEventDeletionCallbackszXPU event recordzCallbackRegistry[int, int]EventRecordCallbackszXPU event waitEventWaitCallbackszXPU memory allocationMemoryAllocationCallbackszXPU memory deallocationMemoryDeallocationCallbackszXPU stream creationStreamCreationCallbackszXPU device synchronizationzCallbackRegistry[[]]DeviceSynchronizationCallbackszXPU stream synchronizationStreamSynchronizationCallbackszXPU event synchronizationEventSynchronizationCallbackscbNreturnc.tj|yN)r add_callbackrs P/var/www/teggl/fontify/venv/lib/python3.12/site-packages/torch/xpu/_gpu_trace.py$register_callback_for_event_creationr ''+c.tj|yr)rrrs r$register_callback_for_event_deletionr$rrc.tj|yr)rrrs r"register_callback_for_event_recordr(s%%b)rc.tj|yr)rrrs r register_callback_for_event_waitr,s##B'rc.tj|yr)r rrs r'register_callback_for_memory_allocationr 0s**2.rc.tj|yr)r rrs r)register_callback_for_memory_deallocationr"4s,,R0rc.tj|yr)r rrs r%register_callback_for_stream_creationr$8s((,rc.tj|yr)r rrs r,register_callback_for_device_synchronizationr&<"//3rc.tj|yr)r rrs r,register_callback_for_stream_synchronizationr)@r'rc.tj|yr)rrrs r+register_callback_for_event_synchronizationr+Ds!..r2r)typingr torch._utilsrr__annotations__rrrr r r r r rintrrrrr r"r$r&r)r+rrr1s )3CCW2X/X2BCW2X/X5E624DDT3U0U5E628H844D40:J : 6;K ; 7:J:6 ,XseTk-B,t,,XseTk-B,t,*8S#J4D+E*$*(3*d2B)C((/#0E/$/1(C5$;2G1D1-hud{.C--4Xb$h5G4D44XseTk5J4t43HcUD[4I3d3r