torch.distributed

Initialization

表1 Initialization

序号

API名称

是否支持

1

torch.distributed.Backend

2

torch.distributed.get_backend

3

torch.distributed.get_rank

4

torch.distributed.get_world_size

5

torch.distributed.init_process_group

6

torch.distributed.is_available

7

torch.distributed.is_initialized

8

torch.distributed.is_mpi_available

9

torch.distributed.is_nccl_available

Distributed Key-Value Store

表2 Distributed Key-Value Store

序号

API名称

是否支持

1

torch.distributed.FileStore

2

torch.distributed.HashStore

3

torch.distributed.PrefixStore

4

torch.distributed.Store

5

torch.distributed.Store.add

6

torch.distributed.Store.delete_key

7

torch.distributed.Store.get

8

torch.distributed.Store.num_keys

9

torch.distributed.Store.set

10

torch.distributed.Store.set_timeout

11

torch.distributed.Store.wait

12

torch.distributed.TCPStore

Groups

表3 Groups

序号

API名称

是否支持

1

torch.distributed.new_group

Point-to-point communication

表4 Point-to-point communication

序号

API名称

是否支持

1

torch.distributed.irecv

2

torch.distributed.isend

3

torch.distributed.recv

4

torch.distributed.send

Synchronous and asynchronous collective operations

表5 Synchronous and asynchronous collective operations

序号

API名称

是否支持

1

is_completed

2

wait

Collective functions

表6 Collective functions

序号

API名称

是否支持

1

torch.distributed.all_gather

2

torch.distributed.all_gather_object

3

torch.distributed.all_reduce

4

torch.distributed.all_to_all

5

torch.distributed.barrier

6

torch.distributed.broadcast

7

torch.distributed.broadcast_object_list

8

torch.distributed.gather

9

torch.distributed.gather_object

10

torch.distributed.reduce

11

torch.distributed.reduce_op

12

torch.distributed.reduce_op

13

torch.distributed.reduce_scatter

14

torch.distributed.ReduceOp

15

torch.distributed.scatter

16

torch.distributed.scatter_object_list

Multi-GPU collective functions

表7 Multi-GPU collective functions

序号

API名称

是否支持

1

torch.distributed.all_gather_multigpu

2

torch.distributed.all_reduce_multigpu

3

torch.distributed.broadcast_multigpu

4

torch.distributed.reduce_multigpu

5

torch.distributed.reduce_scatter_multigpu

Launch utility

表8 Launch utility

序号

API名称

是否支持

1

torch.distributed.launch

Spawn utility

表9 Spawn utility

序号

API名称

是否支持

1

torch.multiprocessing.spawn