U Ç9%eŽ ã@súUddlZddlmZmZmZddlmZe 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<eegdfddœdd „Zeegdfddœd!d"„Zeeegdfddœd#d$„Zeeegdfddœd%d&„Zeegdfddœd'd(„Zeegdfddœd)d*„Zeegdfddœd+d,„Zegdfddœd-d.„Zeegdfddœd/d0„Z eegdfddœd1d2„Z!dS)3éN)ÚCallableÚGenericÚList)Ú ParamSpecÚPc@sHeZdZedœdd„Zeedfddœdd„Zejej ddœd d „Z dS) ÚCallbackRegistry)ÚnamecCs||_g|_dS©N)rÚ callback_list)Úselfr©r úV/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/torch/utils/_cuda_trace.pyÚ__init__ szCallbackRegistry.__init__N©ÚcbÚreturncCs|j |¡dSr )r Úappend)r rr r r Ú add_callbackszCallbackRegistry.add_callback)ÚargsÚkwargsrc OsP|jD]D}z|||ŽWqtk rH}zt d|j¡W5d}~XYqXqdS)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Ú!CUDAEventSynchronizationCallbacksrcCst |¡dSr )rr©rr r r Ú)register_callback_for_cuda_event_creation<sr*cCst |¡dSr )r rr)r r r Ú)register_callback_for_cuda_event_deletion@sr+cCst |¡dSr )r!rr)r r r Ú'register_callback_for_cuda_event_recordDsr,cCst |¡dSr )r"rr)r r r Ú%register_callback_for_cuda_event_waitHsr-cCst |¡dSr )r#rr)r r r Ú,register_callback_for_cuda_memory_allocationLsr.cCst |¡dSr )r$rr)r r r Ú.register_callback_for_cuda_memory_deallocationPsr/cCst |¡dSr )r%rr)r r r Ú*register_callback_for_cuda_stream_creationTsr0cCst |¡dSr )r&rr)r r r Ú1register_callback_for_cuda_device_synchronizationXsr1cCst |¡dSr )r'rr)r r r Ú1register_callback_for_cuda_stream_synchronization\sr2cCst |¡dSr )r(rr)r r r Ú0register_callback_for_cuda_event_synchronizationbsr3)"ÚloggingÚtypingrrrZtyping_extensionsrÚ getLoggerrrrrrÚ__annotations__r r!r"r#r$r%r&r'r(Úintr*r+r,r-r.r/r0r1r2r3r r r r Ús`  ÿ ÿ ÿ ÿ ÿ ÿ ÿ ÿ ÿ ÿ  þ