Esempio n. 1
0
def external_deployment_shards_2(external_deployment_shards_2_args):
    return Deployment(external_deployment_shards_2_args)
Esempio n. 2
0
def external_deployment_pre_shards(external_deployment_pre_shards_args):
    return Deployment(external_deployment_pre_shards_args)
Esempio n. 3
0
def test_cuda_assignment(device_str, replicas, expected):
    os.environ['CUDA_TOTAL_DEVICES'] = str(3)
    actual = Deployment._roundrobin_cuda_device(device_str, replicas)
    assert actual == expected