o ÜÜÓhŽ ã@sUddlZddlmZmZmZddlmZe e¡Z edƒZ Gdd„dee ƒZ e dƒZ de d <e d ƒZde d <e d ƒZd e d<e dƒZd e d<e dƒZde d<e dƒZde d<e dƒZde d<e dƒZde d<e dƒZde d<e dƒZde d<deegdfddfd d!„Zdeegdfddfd"d#„Zdeeegdfddfd$d%„Zdeeegdfddfd&d'„Zdeegdfddfd(d)„Zdeegdfddfd*d+„Zdeegdfddfd,d-„Zdegdfddfd.d/„Zdeegdfddfd0d1„Z deegdfddfd2d3„Z!dS)4éN)ÚCallableÚGenericÚList)Ú ParamSpecÚPc@sNeZdZdefdd„Zdeedfddfdd„Zd ejd ej ddfd d „Z dS) ÚCallbackRegistryÚnamecCs||_g|_dS©N)rÚ callback_list)Úselfr©r úM/var/www/html/ai/venv/lib/python3.10/site-packages/torch/utils/_cuda_trace.pyÚ__init__ s zCallbackRegistry.__init__ÚcbNÚreturncCs|j |¡dSr )r Úappend)r rr r r Ú add_callbackszCallbackRegistry.add_callbackÚargsÚkwargsc OsT|jD]$}z ||i|¤ŽWqty'}z t d|j¡WYd}~qd}~wwdS)Nz7Exception in callback for %s registered with CUDA trace)r Ú ExceptionÚloggerÚ exceptionr)r rrrÚer r r Úfire_callbackss ÿ€ÿýzCallbackRegistry.fire_callbacks) Ú__name__Ú __module__Ú __qualname__Ústrrrrrrrrr r r r r srzCUDA event creationzCallbackRegistry[int]ÚCUDAEventCreationCallbackszCUDA event deletionÚCUDAEventDeletionCallbackszCUDA event recordzCallbackRegistry[int, int]ÚCUDAEventRecordCallbackszCUDA event waitÚCUDAEventWaitCallbackszCUDA memory allocationÚCUDAMemoryAllocationCallbackszCUDA memory deallocationÚCUDAMemoryDeallocationCallbackszCUDA stream creationÚCUDAStreamCreationCallbackszCUDA device synchronizationzCallbackRegistry[[]]Ú"CUDADeviceSynchronizationCallbackszCUDA stream synchronizationÚ"CUDAStreamSynchronizationCallbackszCUDA event synchronizationÚ!CUDAEventSynchronizationCallbacksrrcCót |¡dSr )rr©rr r r Ú)register_callback_for_cuda_event_creation<ór*cCr(r )rrr)r r r Ú)register_callback_for_cuda_event_deletion@r+r,cCr(r )r rr)r r r Ú'register_callback_for_cuda_event_recordDr+r-cCr(r )r!rr)r r r Ú%register_callback_for_cuda_event_waitHr+r.cCr(r )r"rr)r r r Ú,register_callback_for_cuda_memory_allocationLr+r/cCr(r )r#rr)r r r Ú.register_callback_for_cuda_memory_deallocationPr+r0cCr(r )r$rr)r r r Ú*register_callback_for_cuda_stream_creationTr+r1cCr(r )r%rr)r r r Ú1register_callback_for_cuda_device_synchronizationXr+r2cCr(r )r&rr)r r r Ú1register_callback_for_cuda_stream_synchronization\sr3cCr(r )r'rr)r r r Ú0register_callback_for_cuda_event_synchronizationbr+r4)"ÚloggingÚtypingrrrÚtyping_extensionsrÚ getLoggerrrrrrÚ__annotations__rr r!r"r#r$r%r&r'Úintr*r,r-r.r/r0r1r2r3r4r r r r Úsd    ÿ ÿ ÿ ÿ ÿ ÿ ÿ ÿ ÿ ÿ ÿ þ