o i3 @sUddlmZddlmZedZded<edZded<edZd ed <ed Zd ed <ed Z ded<edZ ded<edZ ded<edZ ded<edZ ded<edZded<deegdfddfddZdeegdfddfdd Zdeeegdfddfd!d"Zdeeegdfddfd#d$Zdeegdfddfd%d&Zdeegdfddfd'd(Zdeegdfddfd)d*Zdegdfddfd+d,Zdeegdfddfd-d.Zdeegdfddfd/d0ZdS)1)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 synchronizationEventSynchronizationCallbackscbNreturncCt|dSN)r add_callbackrrQ/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/torch/xpu/_gpu_trace.py$register_callback_for_event_creation rcCrr)rrrrrr$register_callback_for_event_deletion$rrcCrr)rrrrrr"register_callback_for_event_record(rrcCrr)rrrrrr register_callback_for_event_wait,rrcCrr)rrrrrr'register_callback_for_memory_allocation0rrcCrr)r rrrrr)register_callback_for_memory_deallocation4rrcCrr)r rrrrr%register_callback_for_stream_creation8rrcCrr)r rrrrr,register_callback_for_device_synchronization<rrcCrr)r rrrrr,register_callback_for_stream_synchronization@rrcCrr)r rrrrr+register_callback_for_event_synchronizationDrr )typingr torch._utilsrr__annotations__rrrrr r r r r intrrrrrrrrrr rrrrsH