קg@GddZeZdZdZdS)c>eZdZdZdZdZdZdZdZdZ dZ d S) CompilationCallbackHandlerc"g|_g|_dS)N)start_callbacks end_callbacksselfs R/var/www/html/ai-engine/env/lib/python3.11/site-packages/torch/_dynamo/callback.py__init__z#CompilationCallbackHandler.__init__s!c:|j||S)z Register a callback function to be called when the compilation starts. Args: - callback (callable): The callback function to register. )rappendrcallbacks r register_start_callbackz2CompilationCallbackHandler.register_start_callbacks  ##H---r c:|j||S)z Register a callback function to be called when the compilation ends. Args: - callback (callable): The callback function to register. )rr rs r register_end_callbackz0CompilationCallbackHandler.register_end_callbacks  !!(+++r c:|j|dS)z Remove a registered start callback function. Args: - callback (callable): The callback function to remove. N)rremovers r remove_start_callbackz0CompilationCallbackHandler.remove_start_callbacks! ##H-----r c:|j|dS)z Remove a registered end callback function. Args: - callback (callable): The callback function to remove. N)rrrs r remove_end_callbackz.CompilationCallbackHandler.remove_end_callback$s! !!(+++++r c.|jD] }| dS)z9 Execute all registered start callbacks. N)rrs r run_start_callbacksz.CompilationCallbackHandler.run_start_callbacks-s-,  H HJJJJ  r c.|jD] }| dS)z7 Execute all registered end callbacks. N)rrs r run_end_callbacksz,CompilationCallbackHandler.run_end_callbacks4s-*  H HJJJJ  r cj|j|jdS)z1 Clear all registered callbacks. N)rclearrrs r rz CompilationCallbackHandler.clear;s4 ""$$$   """""r N) __name__ __module__ __qualname__r rrrrrrrr r rrs   ...,,,#####r rc:t||S)zU Decorator to register a callback function for the start of the compilation. )callback_handlerrrs r on_compile_startr%Fs,,X666 Or c:t||S)zS Decorator to register a callback function for the end of the compilation. )r#rr$s r on_compile_endr'Ns**8444 Or N)rr#r%r'r!r r r(sg>#>#>#>#>#>#>#>#B.-//r