o h@sTUdZddlmZddlmZddlmZdaeej e d<deej fdd Z dS) au Manages process groups for distributed compilation in TorchDynamo. This module handles the initialization and management of process groups used for distributed compilation. Key features: - Lazy initialization of compilation process groups - Only creates groups when distributed mode is enabled and available - Integrates with compiler_collectives configuration setting - Provides a single global process group for compilation coordination The process group is created only when needed and if the distributed environment is properly initialized, making it safe to import and use this module even in non-distributed scenarios. )OptionalN)config _COMPILE_PGreturncCs4tjrtrtrtdurtjjddatSdS)Npt2_compile_pg)pg_tag)renable_compiler_collectivesdist is_availableis_initializedrdistributed_c10d_new_group_with_tagrrm/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/torch/_dynamo/distributed.pyget_compile_pgsr) __doc__typingrtorch.distributed distributedr rr ProcessGroup__annotations__rrrrrs