Method CreateCudaGPUs

src/core/device/platform.cc:114–120

}

const vector<shared_ptr<Device>> Platform::CreateCudaGPUs(
    const size_t num_devices, size_t init_size) {
  const vector<int> gpus = GetGPUIDs();
  CHECK_LE(num_devices, gpus.size());
  vector<int> use_gpus(gpus.begin(), gpus.begin() + num_devices);
  return CreateCudaGPUsOn(use_gpus, init_size);
}

const vector<shared_ptr<Device>> Platform::CreateCudaGPUsOn(
    const vector<int>& devices, size_t init_size) {