def external_deployment_shards_2(external_deployment_shards_2_args): return Deployment(external_deployment_shards_2_args)
def external_deployment_pre_shards(external_deployment_pre_shards_args): return Deployment(external_deployment_pre_shards_args)
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