a ¢º”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<eegdfddœdd„Zeegdfddœdd„Zeeegdfddœd d!„Zeeegdfddœd"d#„Zeegdfddœd$d%„Zeegdfddœd&d'„Zeegdfddœd(d)„Zegdfddœd*d+„Zeegdfddœd,d-„Zeegdfddœd.d/„ZdS)0é)Ú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ÚEventSynchronizationCallbacksN)ÚcbÚreturncCst |¡dS©N)rÚ add_callback©r©rúB/var/www/auris/lib/python3.9/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_wait,srcCst |¡dSr)rrrrrrÚ'register_callback_for_memory_allocation0srcCst |¡dSr)r rrrrrÚ)register_callback_for_memory_deallocation4srcCst |¡dSr)r rrrrrÚ%register_callback_for_stream_creation8srcCst |¡dSr)r rrrrrÚ,register_callback_for_device_synchronization<srcCst |¡dSr)r rrrrrÚ,register_callback_for_stream_synchronization@srcCst |¡dSr)r rrrrrÚ+register_callback_for_event_synchronizationDsr)ÚtypingrZ torch._utilsrrÚ__annotations__rrrrr r r r r ÚintrrrrrrrrrrrrrrÚsF ÿ ÿ ÿ ÿ ÿ ÿ ÿ