+ jRt^RIHt^RIHt^RIHt^RIH t ^RI H t ^RI H t ^RIHtHtHt^RIHtHt^R IHt^R IHt^R IHtR R ltRRltRRlt]!4RR3RRllt]!4R]R]R3RRlltR#)a Factory functions for creating checkpointer instances with sensible defaults. This module provides high-level factory functions that simplify the creation of checkpointer instances by automatically handling component initialization and configuration with reasonable defaults. )CallableAnyN)create_barrier_from_config)CheckpointProcess)CheckpointReader)CheckpointWriterCheckpointWriterConfig WriterHook)AsyncCheckpointerSyncCheckpointer)CheckpointerConfig) DefaultStagerRankInfoc$V^8dQhR\/#)returnr)formats"s/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/distributed/checkpoint/_experimental/builder.py __annotate__rs<<<c\P!4'd4\\P!4\P!4R7#\^^R7#)z Get default rank information from the current distributed environment. Returns: RankInfo: Rank information from the default process group if initialized, otherwise single-rank fallback. )global_world_size global_rank)distis_initializedrget_world_sizeget_rankrr_get_default_rank_infor sD "113   !;;rc(V^8dQhR\RR/#)r_rNr)rs"rrr*s>>3>4>rcR#)z3Default subprocess initialization function (no-op).Nr)r"s*rdefault_subprocess_init_fnr$*src0V^8dQhR\R\/#)r rank_infor)rr)rs"rrr.sh3Crc,\\4VR7#)z2Default checkpoint writer initialization function.)configr&)rr r&s&rdefault_writer_init_fnr*.s %' rcdV^8dQhR\R\R,R\R,R\/#)rr(r&N commit_hookr)r rr r )rs"rrr6s988 8$8d"8 8rcVf \4p\VR7p\VP4p\ VP VVVR7p\ VVR7#)a- Factory function to create a SyncCheckpointer instance with sensible defaults. This function creates a synchronous checkpointer with default components, automatically detecting rank information from the default process group if available, and using the provided component configurations. Args: config: CheckpointerConfig containing component-specific configurations (writer_config, staging_config, process_config). Defaults to CheckpointerConfig(). rank_info: RankInfo for distributed training. Defaults to auto-detection from the default PyTorch distributed process group if initialized, otherwise falls back to single-rank (world_size=1, rank=0). commit_hook: Optional hook for custom actions before and after checkpoint commits. Returns: SyncCheckpointer: A configured synchronous checkpointer instance. Examples: # Simplest usage - auto-detect rank, default config checkpointer = make_sync_checkpointer() # Explicit rank configuration checkpointer = make_sync_checkpointer( rank_info=RankInfo(global_world_size=4, global_rank=0) ) # Disable barrier from .barriers import BarrierConfig config = CheckpointerConfig(barrier_config=BarrierConfig(barrier_type=None)) checkpointer = make_sync_checkpointer(config=config) r))r(r&barrierr,)writerreader)r rrbarrier_configr writer_configr )r(r&r,r0r.r/s&&& rmake_sync_checkpointerr36scJ*, F))>)>?G ## F  rcV^8dQhR\R\R,R\R ,R\\R3,R\R\ 3,R\ \\3,R,R \/#) rr(r&Nsubprocess_init_fn.subprocess_init_argscheckpoint_writer_init_fncheckpoint_writer_init_argsr).N) r rrtuplerrdictstrr )rs"rrrqsw== =$=!+= S/ = (-=(=> = "&c3h$!6 ==rc Vf \4p\VR7p\VPR7pT;'g/p\ VVP VVVVR7p\ VVVR7#)a# Factory function to create an AsyncCheckpointer instance with sensible defaults. This function creates an asynchronous checkpointer using the provided configuration, automatically detecting rank information if not provided. Args: config: CheckpointerConfig containing component-specific configurations. rank_info: RankInfo for distributed training. Defaults to auto-detection. subprocess_init_fn: Function to initialize the subprocess. Defaults to no-op. subprocess_init_args: Arguments to pass to subprocess_init_fn. checkpoint_writer_init_fn: Function to create CheckpointWriter instance. checkpoint_writer_init_args: Arguments to pass to checkpoint_writer_init_fn. Returns: AsyncCheckpointer: A configured asynchronous checkpointer instance. Examples: # Create with default config checkpointer = make_async_checkpointer() # Create with custom init functions checkpointer = make_async_checkpointer( subprocess_init_fn=my_subprocess_init_fn, checkpoint_writer_init_fn=my_writer_init_fn ) r))r()r&r(r5r6r7r8)checkpoint_stagercheckpoint_processr0)r rrstaging_configrprocess_configr ) r(r&r5r6r7r8r0r=r>s &&&&&& rmake_async_checkpointerrAqsF*, F&$$#>"C"C*$$-1";$?  +- rr) __doc__collections.abcrtypingrtorch.distributed distributedrbarriersrr>rcheckpoint_readerrcheckpoint_writerrr r checkpointerr r r(r stagingrtypesrr r$r*r3rArrrrMst% 01/SS=&"<$>"4!5!%%)8x"4!5!%.H,.AW9= ==r