U MhE @sUddlmZddlmZedZded<edZded<edZd ed <ed Zd ed <ed Z ded<edZ ded<edZ ded<edZ ded<edZ ded<edZded<eegdfddddZeegdfddddZeeegdfddd d!Zeeegdfddd"d#Zeegdfddd$d%Zeegdfddd&d'Zeegdfddd(d)Zegdfddd*d+Zeegdfddd,d-Zeegdfddd.d/ZdS)0)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 synchronizationEventSynchronizationCallbacksN)cbreturncCst|dSN)r add_callbackrrF/var/www/html/venv/lib/python3.8/site-packages/torch/xpu/_gpu_trace.py$register_callback_for_event_creation&srcCst|dSr)rrrrrr$register_callback_for_event_deletion*srcCst|dSr)rrrrrr"register_callback_for_event_record.srcCst|dSr)rrrrrr register_callback_for_event_wait2srcCst|dSr)rrrrrr'register_callback_for_memory_allocation6srcCst|dSr)r rrrrr)register_callback_for_memory_deallocation:srcCst|dSr)r rrrrr%register_callback_for_stream_creation>srcCst|dSr)r rrrrr,register_callback_for_device_synchronizationBsrcCst|dSr)r rrrrr,register_callback_for_stream_synchronizationFsrcCst|dSr)r rrrrr+register_callback_for_event_synchronizationJsr)typingrZ torch._utilsrr__annotations__rrrrr r r r r intrrrrrrrrrrrrrrsT