Ig2GddZeZdZdZy)c<eZdZdZdZdZdZdZdZdZ dZ y ) CompilationCallbackHandlerc g|_g|_y)N)start_callbacks end_callbacksselfs S/var/www/html/answerous/venv/lib/python3.12/site-packages/torch/_dynamo/callback.py__init__z#CompilationCallbackHandler.__init__s!c<|jj||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<|jj||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:|jj|y)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:|jj|y)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 c4|jD] }| y)z9 Execute all registered start callbacks. N)rrs r run_start_callbacksz.CompilationCallbackHandler.run_start_callbacks-s,, H J r c4|jD] }| y)z7 Execute all registered end callbacks. N)rrs r run_end_callbacksz,CompilationCallbackHandler.run_end_callbacks4s** H J r cl|jj|jjy)z1 Clear all registered callbacks. N)rclearrrs r rz CompilationCallbackHandler.clear;s( ""$   "r N) __name__ __module__ __qualname__r rrrrrrrr r rrs* .,#r rc0tj||S)zU Decorator to register a callback function for the start of the compilation. )callback_handlerrrs r on_compile_startr%Fs,,X6 Or c0tj||S)zS Decorator to register a callback function for the end of the compilation. )r#rr$s r on_compile_endr'Ns**84 Or N)rr#r%r'r!r r r(s%>#>#B./r