| # Copyright (c) Meta Platforms, Inc. and affiliates. | |
| # | |
| # This software may be used and distributed in accordance with | |
| # the terms of the DINOv3 License Agreement. | |
| # isort: skip_file | |
| from .torch_distributed_wrapper import ( | |
| disable_distributed as disable, | |
| enable_distributed as enable, | |
| get_default_process_group, | |
| get_process_subgroup, | |
| get_rank, | |
| get_subgroup_rank, | |
| get_subgroup_size, | |
| get_world_size, | |
| is_distributed_enabled as is_enabled, | |
| is_main_process, | |
| is_subgroup_main_process, | |
| new_subgroups, | |
| save_in_main_process, | |
| TorchDistributedEnvironment, | |
| ) | |
| from .torch_distributed_primitives import gather_all_tensors, reduce_dict | |