+ jRt^RIt^RIt^RIHtHt^RIHt^RIH t H t ^RI H t !RR]P4t] !RR 44t] !R R 44t]!4tR R ltRRltR#)a  This module provides callback management functionality for TorchDynamo's compilation process. It implements a thread-safe system for registering, managing and executing callbacks that run at the start and end of TorchDynamo compilations. Key features include: - Registration and deregistration of compilation callbacks - Thread-safe callback handling with proper locking mechanisms - Prevention of duplicate callback execution when configured - Decorator utilities for easy callback registration - Context manager for controlled callback lifecycle The module centers around the CompilationCallbackHandler class which maintains separate lists for start and end callbacks, manages their execution order, and ensures thread-safety. Utility decorators @on_compile_start and @on_compile_end provide a convenient way to register compilation hooks. Example usage: @on_compile_start def my_start_callback(): print("Starting compilation") @on_compile_end def my_end_callback(): print("Compilation complete") N)Callable Generator)contextmanager) dataclassfield)Anyc&]tRt^$t^t^t^t^tRtR#)CallbackTriggerN) __name__ __module__ __qualname____firstlineno__DYNAMO LAZY_BACKWARDTRITON_AUTOTUNINGCUDAGRAPH_RECORDING__static_attributes__r W/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/_dynamo/callback.pyr r $s FMrr c,a]tRt^/toV3RltRtVtR#) CallbackArgsc2<V^8dQh/S[;R&S[;R&#)callback_trigger compile_id)r str)format __classdict__s"r __annotate__CallbackArgs.__annotate__/s%%Orr N)r r r r__annotate_func__r__classdictcell__rs@rrr/srrc*a]tRt^5to]!]R7t]!]R7t]!^RRR7t]!] PRRR7t V3RlRlt V3RlRlt V3R lR ltV3R lR ltV3R lRltV3RlRlt]V3RlRl4tV3RlRltV3RltRtVtR#)CompilationCallbackHandler)default_factoryF)defaultinitrepr)r&r(r)cR<V^8dQhRS[S[.R3,RS[S[.R3,/#rcallbackNreturnrr)rrs"rr'CompilationCallbackHandler.__annotate__?5   ,!56  <.$& ' rc <VPPV4V#)z Register a callback function to be called when the compilation starts. Args: - callback (Callable): The callback function to register. )start_callbacksappendselfr,s&&rregister_start_callback2CompilationCallbackHandler.register_start_callback?s ##H-rcR<V^8dQhRS[S[.R3,RS[S[.R3,/#r+r.)rrs"rrr/Kr0rc <VPPV4V#)z Register a callback function to be called when the compilation ends. Args: - callback (Callable): The callback function to register. ) end_callbacksr3r4s&&rregister_end_callback0CompilationCallbackHandler.register_end_callbackKs !!(+rc:<V^8dQhRS[S[.R3,RR/#r+r.)rrs"rrr/Ws(..h ~t7K.L.QU.rc <VPPV4R#)zm Remove a registered start callback function. Args: - callback (Callable): The callback function to remove. N)r2remover4s&&rremove_start_callback0CompilationCallbackHandler.remove_start_callbackWs ##H-rc:<V^8dQhRS[S[.R3,RR/#r+r.)rrs"rrr/`s',,Hl^T5I,J,t,rc <VPPV4R#)zk Remove a registered end callback function. Args: - callback (Callable): The callback function to remove. N)r:r?r4s&&rremove_end_callback.CompilationCallbackHandler.remove_end_callback`s !!(+rc$<V^8dQhRS[RR/#rargsr-Nr)rrs"rrr/is rc <VPF pV!V4K R#)z) Execute all registered start callbacks. N)r2r5rHr,s&& rrun_start_callbacks.CompilationCallbackHandler.run_start_callbacksis,,H TN-rc$<V^8dQhRS[RR/#rGrI)rrs"rrr/psltrc <VPF pV!V4K R#)z' Execute all registered end callbacks. N)r:rKs&& rrun_end_callbacks,CompilationCallbackHandler.run_end_callbacksps**H TN+rc D<V^8dQhRS[RS[RS[RS[S[3,/#)rtriggerrr-N)r rrr)rrs"rrr/xs066&6476 4c> "6rc # ,"\W4pVP;_uu_4V;P^, unVP^8XdVPV4RRR4RxVP;_uu_4VP^8:d \ R4hVP^8XdVP V4V;P^,unRRR4R# +'giL;i +'giR#;i TP;_uu_4TP^8:d \ R4hTP^8XdTP T4T;P^,unRRR4i +'gii;i;i5i)zS Context manager to install the callbacks and run them when the context is exited. Nz1Pending callbacks counter cannot become negative.)r;_CompilationCallbackHandler__pending_callbacks_counter_lock6_CompilationCallbackHandler__pending_callbacks_counterrLAssertionErrorrP)r5rSrrHs&&& rinstall_callbacks,CompilationCallbackHandler.install_callbacksws5G0 666600A5033q8,,T27 66633q8(K33q8**4000A5076 76 76666633q8(K33q8**4000A507666se FD 5 F> F FFc<V^8dQhRR/#)rr-Nr )rrs"rrr/s  t rc VPP4VPP4VP^8wd\ RVP R24hR#)z! Clear all registered callbacks. zCannot clear callbacks while z are pendingN)r2clearr:rVrW)r5s&rr\ CompilationCallbackHandler.clears\ ""$   "  + +q 0 /0P0P/QQ]^  1rc<V^8dQh/S[S[S[.R3,,;R&S[S[S[.R3,,;R&S[;R&S[P ;R&#)rNr2r:rVrU)listrrint threadingLock)rrs"rrr/5s_(L>4#789W,!567U "%O  '0nn rr N)r r r rrr_r2r:rVrarbrUr6r;r@rDrLrPrrXr\r!rr"r#s@rr%r%5srssy6 /%(dii    bb bJ./r