Add initP2P

helinwang-patch-1
Yu Yang 7 years ago
parent 50f71f5005
commit dcf7bd2d92

@ -85,7 +85,7 @@ void InitDevices() {
for (int i = 0; i < count; ++i) { for (int i = 0; i < count; ++i) {
places.emplace_back(platform::CUDAPlace(i)); places.emplace_back(platform::CUDAPlace(i));
} }
// InitP2P(count); InitP2P(count);
platform::DeviceContextPool::Init(places); platform::DeviceContextPool::Init(places);
} }

Loading…
Cancel
Save