o "i@s(GdddZeZddZddZdS)c@sLeZdZddZddZddZddZd d Zd d Zd dZ ddZ dS)CompilationCallbackHandlercCsg|_g|_dS)N)start_callbacks end_callbacksselfrT/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/torch/_dynamo/callback.py__init__s z#CompilationCallbackHandler.__init__cC|j||S)z Register a callback function to be called when the compilation starts. Args: - callback (callable): The callback function to register. )rappendrcallbackrrrregister_start_callback z2CompilationCallbackHandler.register_start_callbackcCr )z Register a callback function to be called when the compilation ends. Args: - callback (callable): The callback function to register. )rr r rrrregister_end_callbackrz0CompilationCallbackHandler.register_end_callbackcC|j|dS)z Remove a registered start callback function. Args: - callback (callable): The callback function to remove. N)rremover rrrremove_start_callbackz0CompilationCallbackHandler.remove_start_callbackcCr)z Remove a registered end callback function. Args: - callback (callable): The callback function to remove. N)rrr rrrremove_end_callback$rz.CompilationCallbackHandler.remove_end_callbackcC|jD]}|qdS)z9 Execute all registered start callbacks. N)rr rrrrun_start_callbacks- z.CompilationCallbackHandler.run_start_callbackscCr)z7 Execute all registered end callbacks. N)rr rrrrun_end_callbacks4rz,CompilationCallbackHandler.run_end_callbackscCs|j|jdS)z1 Clear all registered callbacks. N)rclearrrrrrr;s z CompilationCallbackHandler.clearN) __name__ __module__ __qualname__rr rrrrrrrrrrrs     rcCt||S)zU Decorator to register a callback function for the start of the compilation. )callback_handlerr r rrron_compile_startF r cCr)zS Decorator to register a callback function for the end of the compilation. )rrrrrron_compile_endNr!r"N)rrr r"rrrrsA