IЦi SSKJr SSKJrJr SSKJrJrJrJ r \"SS55r \ "5r S\/S4S\/S44S jr S\/S4S\/S44S jr g) )contextmanager) dataclassfield)AnyCallable GeneratorListc(\rSrSr%\"\S9r\\/S4\ S'\"\S9r \\/S4\ S'S\/S4S\/S44Sjr S\/S4S\/S44S jr S\/S4SS4S jr S\/S4SS4S jrSS jrSS jr\S\S\\44Sj5rSSjrSrg)CompilationCallbackHandler)default_factoryNstart_callbacks end_callbackscallbackreturnc<URRU5 U$)z Register a callback function to be called when the compilation starts. Args: - callback (Callable): The callback function to register. )rappendselfrs U/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/torch/_dynamo/callback.pyregister_start_callback2CompilationCallbackHandler.register_start_callback s ##H-c<URRU5 U$)z Register a callback function to be called when the compilation ends. Args: - callback (Callable): The callback function to register. )rrrs rregister_end_callback0CompilationCallbackHandler.register_end_callbacks !!(+rc:URRU5 g)zm Remove a registered start callback function. Args: - callback (Callable): The callback function to remove. N)rremovers rremove_start_callback0CompilationCallbackHandler.remove_start_callback!s ##H-rc:URRU5 g)zk Remove a registered end callback function. Args: - callback (Callable): The callback function to remove. N)rrrs rremove_end_callback.CompilationCallbackHandler.remove_end_callback*s !!(+rc8URH nU"5 M g)z) Execute all registered start callbacks. N)rrs rrun_start_callbacks.CompilationCallbackHandler.run_start_callbacks3s,,H J-rc8URH nU"5 M g)z' Execute all registered end callbacks. N)rrs rrun_end_callbacks,CompilationCallbackHandler.run_end_callbacks:s**H J+rc## UR5 Sv UR5 g!UR5 f=f7f)zS Context manager to install the callbacks and run them when the context is exited. N)r%r(rs rinstall_callbacks,CompilationCallbackHandler.install_callbacksAs1  %  $ $ &   " " $D " " $s>)>;>clURR5 URR5 g)z! Clear all registered callbacks. N)rclearrr+s rr/ CompilationCallbackHandler.clearLs( ""$   "r)rN)__name__ __module__ __qualname____firstlineno__rlistrr r__annotations__rrrrr"r%r(rrrr,r/__static_attributes__r1rrr r s05d0KOT(2t8,-K.3D.IM4T*+I  T*  "d(  hr4x.@XbRVhEW.hr4x.@.T.,HRX,>,4,%9T3^#<%%#rr rNrc0[RU5 U$)zM Decorator to register a callback function for the start of the compilation. )callback_handlerrrs ron_compile_startr<Ws,,X6 Orc0[RU5 U$)zK Decorator to register a callback function for the end of the compilation. )r:rr;s ron_compile_endr>_s**84 Or) contextlibr dataclassesrrtypingrrrr r r:r<r>r1rrrBs%(11 J#J# J#Z./xD1hr4x6HXb$h/HRX4Fr