o vZŽh3 ã@sØUddlmZddlmZedƒZded<edƒZded<edƒZd ed <ed ƒZd ed <ed ƒZ ded<edƒZ ded<edƒZ ded<edƒZ ded<edƒZ ded<edƒZded<deegdfddfdd„Zdeegdfddfdd „Zdeeegdfddfd!d"„Zdeeegdfddfd#d$„Zdeegdfddfd%d&„Zdeegdfddfd'd(„Zdeegdfddfd)d*„Zdegdfddfd+d,„Zdeegdfddfd-d.„Zdeegdfddfd/d0„ZdS)1é)ÚCallable)ÚCallbackRegistryzXPU event creationzCallbackRegistry[int]ÚEventCreationCallbackszXPU event deletionÚEventDeletionCallbackszXPU event recordzCallbackRegistry[int, int]ÚEventRecordCallbackszXPU event waitÚEventWaitCallbackszXPU memory allocationÚMemoryAllocationCallbackszXPU memory deallocationÚMemoryDeallocationCallbackszXPU stream creationÚStreamCreationCallbackszXPU device synchronizationzCallbackRegistry[[]]ÚDeviceSynchronizationCallbackszXPU stream synchronizationÚStreamSynchronizationCallbackszXPU event synchronizationÚEventSynchronizationCallbacksÚcbNÚreturncCót |¡dS©N)rÚ add_callback©r©rúC/var/www/auris/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 )ÚtypingrZ torch._utilsrrÚ__annotations__rrrrr r r r r Úintrrrrrrrrrr rrrrÚsH  ÿ ÿ ÿ ÿ ÿ ÿ ÿ