Exemplos de _test_distrib_broadcast em Python

Exemplo n.º 1

def test_idist_collective_ops_no_dist():
    _test_distrib_all_reduce("cpu")
    _test_distrib_all_gather("cpu")
    _test_distrib_barrier("cpu")
    _test_distrib_broadcast("cpu")

    if torch.cuda.device_count() > 1:
        _test_distrib_all_reduce("cuda")
        _test_distrib_all_gather("cuda")
        _test_distrib_barrier("cuda")
        _test_distrib_broadcast("cuda")

Exemplo n.º 2

def test_idist_broadcast_gloo(distributed_context_single_node_gloo):

    device = "cpu"
    _test_distrib_broadcast(device)

Exemplo n.º 3

def test_idist_broadcast_nccl(distributed_context_single_node_nccl):

    device = "cuda:{}".format(
        distributed_context_single_node_nccl["local_rank"])
    _test_distrib_broadcast(device)

Exemplo n.º 4

Arquivo: test_native.py Projeto: MinjaMiladinovic/ignite

def test_idist_broadcast_nccl(distributed_context_single_node_nccl):

    device = f"cuda:{distributed_context_single_node_nccl['local_rank']}"
    _test_distrib_broadcast(device)

Exemplo n.º 5

Arquivo: test_xla.py Projeto: isabella232/ignite-2

def _test_idist_broadcast_xla_in_child_proc(index):
    device = idist.device()
    _test_distrib_broadcast(device)

Exemplo n.º 6

Arquivo: test_xla.py Projeto: isabella232/ignite-2

def test_idist_broadcast_xla():

    device = idist.device()
    _test_distrib_broadcast(device)

Exemplo n.º 7

Arquivo: test_native.py Projeto: vfdev-5/ignite

def test_idist_broadcast_nccl(distributed_context_single_node_nccl):

    device = idist.device()
    _test_distrib_broadcast(device)