U zh@s(GdddZeZddZddZdS)c@sLeZdZddZddZddZddZd d Zd d Zd dZ ddZ dS)CompilationCallbackHandlercCsg|_g|_dS)N)start_callbacks end_callbacksselfrH/var/www/html/venv/lib/python3.8/site-packages/torch/_dynamo/callback.py__init__sz#CompilationCallbackHandler.__init__cCs|j||S)z Register a callback function to be called when the compilation starts. Args: - callback (callable): The callback function to register. )rappendrcallbackrrrregister_start_callbacks z2CompilationCallbackHandler.register_start_callbackcCs|j||S)z Register a callback function to be called when the compilation ends. Args: - callback (callable): The callback function to register. )rr r rrrregister_end_callbacks z0CompilationCallbackHandler.register_end_callbackcCs|j|dS)z Remove a registered start callback function. Args: - callback (callable): The callback function to remove. N)rremover rrrremove_start_callbacksz0CompilationCallbackHandler.remove_start_callbackcCs|j|dS)z Remove a registered end callback function. Args: - callback (callable): The callback function to remove. N)rrr rrrremove_end_callback$sz.CompilationCallbackHandler.remove_end_callbackcCs|jD] }|qdS)z9 Execute all registered start callbacks. N)rr rrrrun_start_callbacks-s z.CompilationCallbackHandler.run_start_callbackscCs|jD] }|qdS)z7 Execute all registered end callbacks. N)rr rrrrun_end_callbacks4s z,CompilationCallbackHandler.run_end_callbackscCs|j|jdS)z1 Clear all registered callbacks. N)rclearrrrrrr;s z CompilationCallbackHandler.clearN) __name__ __module__ __qualname__rr r rrrrrrrrrrs    rcCst||S)zU Decorator to register a callback function for the start of the compilation. )callback_handlerr r rrron_compile_startFs rcCst||S)zS Decorator to register a callback function for the end of the compilation. )rr rrrron_compile_endNs rN)rrrrrrrrsA