o z“©io ã@s$ddlmZmZGdd„deƒZdS)é)ÚABCÚabstractmethodc@süeZdZdZeedefdd„ƒƒZeedefdd„ƒƒZ eede fdd„ƒƒZ e edefd d „ƒƒZ ede fd d „ƒZed e ddfdd„ƒZede fdd„ƒZede ddfdd„ƒZede fdd„ƒZede fdd„ƒZde de ddfdd„Zd dd„ZdS)!ÚClusterEnvironmentz'Specification of a cluster environment.ÚreturncCódS)z8Whether the environment creates the subprocesses or not.N©©Úselfrrúm/home/ubuntu/.local/lib/python3.10/site-packages/lightning_fabric/plugins/environments/cluster_environment.pyÚcreates_processes_externallyóz/ClusterEnvironment.creates_processes_externallycCr)zEThe main address through which all processes connect and communicate.Nrrrrr Ú main_addressr zClusterEnvironment.main_addresscCr)zUAn open and configured port in the main node through which all processes communicate.Nrrrrr Ú main_portr zClusterEnvironment.main_portcCr)zbDetects the environment settings corresponding to this cluster and returns ``True`` if they match.Nrrrrr Údetect#r zClusterEnvironment.detectcCr)z5The number of processes across all devices and nodes.Nrrrrr Ú world_size(r zClusterEnvironment.world_sizeÚsizeNcCódS©Nr)r rrrr Úset_world_size,óz!ClusterEnvironment.set_world_sizecCr)zOThe rank (index) of the currently running process across all nodes and devices.Nrrrrr Ú global_rank0r zClusterEnvironment.global_rankÚrankcCrrr)r rrrr Úset_global_rank4rz"ClusterEnvironment.set_global_rankcCr)zMThe rank (index) of the currently running process inside of the current node.Nrrrrr Ú local_rank8r zClusterEnvironment.local_rankcCr)z?The rank (index) of the node on which the current process runs.Nrrrrr Ú node_rank<r zClusterEnvironment.node_rankÚ num_devicesÚ num_nodescCr)z‚Validates settings configured in the script against the environment, and raises an exception if there is an inconsistency.Nr)r rrrrr Úvalidate_settings@sz$ClusterEnvironment.validate_settingscCr)z0Clean up any state set after execution finishes.Nrrrrr ÚteardownErzClusterEnvironment.teardown)rN)Ú__name__Ú __module__Ú __qualname__Ú__doc__ÚpropertyrÚboolr Ústrr ÚintrÚ staticmethodrrrrrrrrrrrrr rs8rN)Úabcrrrrrrr Ús