a ‘º”hI ã@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)ÚCallbackRegistryzCUDA event creationzCallbackRegistry[int]ÚEventCreationCallbackszCUDA event deletionÚEventDeletionCallbackszCUDA event recordzCallbackRegistry[int, int]ÚEventRecordCallbackszCUDA event waitÚEventWaitCallbackszCUDA memory allocationÚMemoryAllocationCallbackszCUDA memory deallocationÚMemoryDeallocationCallbackszCUDA stream creationÚStreamCreationCallbackszCUDA device synchronizationzCallbackRegistry[[]]ÚDeviceSynchronizationCallbackszCUDA stream synchronizationÚStreamSynchronizationCallbackszCUDA event synchronizationÚEventSynchronizationCallbacksN)ÚcbÚreturncCst |¡dS©N)rÚ add_callback©r©rúC/var/www/auris/lib/python3.9/site-packages/torch/cuda/_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_wait0srcCst |¡dSr)rrrrrrÚ'register_callback_for_memory_allocation4srcCst |¡dSr)r rrrrrÚ)register_callback_for_memory_deallocation8srcCst |¡dSr)r rrrrrÚ%register_callback_for_stream_creation<srcCst |¡dSr)r rrrrrÚ,register_callback_for_device_synchronization@srcCst |¡dSr)r rrrrrÚ,register_callback_for_stream_synchronizationDsrcCst |¡dSr)r rrrrrÚ+register_callback_for_event_synchronizationHsr)ÚtypingrZ torch._utilsrrÚ__annotations__rrrrr r r r r ÚintrrrrrrrrrrrrrrÚsN ÿ ÿ ÿ ÿ ÿ ÿ ÿ ÿ ÿ