Method GetGPUIDs

src/core/device/platform.cc:47–56

Return the total num of free GPUs

/// Return the total num of free GPUs
const vector<int> Platform::GetGPUIDs() {
  vector<int> gpus;
  int count = Platform::GetNumGPUs();
  for (int i = 0; i < count; i++) {
    if (Platform::CheckDevice(i)) {
      gpus.push_back(i);
    }
  }
  return gpus;
}

const std::pair<size_t, size_t> Platform::GetGPUMemSize(const int device) {
  std::pair<size_t, size_t> ret{0, 0};