ksuit.distributed.config

Submodules

• ksuit.distributed.config.distributed_config

Functions

set_config(new_config)
is_managed()
get_local_rank()
get_num_nodes()
get_managed_world_size()
get_managed_rank()
is_distributed()
get_rank()
get_world_size()
is_data_rank0()
is_rank0()
is_local_rank0()
barrier()

Package Contents

ksuit.distributed.config.set_config(new_config)

Parameters:

new_config (distributed_config.DistributedConfig)

ksuit.distributed.config.is_managed()

ksuit.distributed.config.get_local_rank()

ksuit.distributed.config.get_num_nodes()

ksuit.distributed.config.get_managed_world_size()

ksuit.distributed.config.get_managed_rank()

ksuit.distributed.config.is_distributed()

ksuit.distributed.config.get_rank()

ksuit.distributed.config.get_world_size()

ksuit.distributed.config.is_data_rank0()

ksuit.distributed.config.is_rank0()

ksuit.distributed.config.is_local_rank0()

ksuit.distributed.config.barrier()