Commit 9d6e47f0 authored by ThomasNing's avatar ThomasNing
Browse files

add the mesh connection to the executable

parent b06330d0
......@@ -284,6 +284,13 @@ struct AllocateAndTransferFunctor
// Transfer data to device
device_mem.ToDevice(host_tensor.data());
int worldSize = arg_parser.get_int("gpu_nums");
setupConnection(device_id,
static_cast<int>(host_gpu),
static_cast<int>(worldSize),
device_mem.GetDeviceBuffer(),
host_tensor.get_element_space_size_in_bytes());
int n_warmup = arg_parser.get_int("warmup");
int n_repeat = arg_parser.get_int("repeat");
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment