get_nvidia_compute_capability.h File Reference

#include <cassert>
#include <stdexcept>
#include <string>
#include <vector>
#include <cuda_runtime.h>

Include dependency graph for get_nvidia_compute_capability.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Functions
std::vector< size_t >	get_nvidia_compute_capability ()

Function Documentation

std::vector<size_t> get_nvidia_compute_capability ( )

inline

Definition at line 26 of file get_nvidia_compute_capability.h.

References to_string().

Referenced by main().

                                                         {
  using namespace std::string_literals;
  std::vector<size_t> ret;

  int deviceCount = 0;
  cudaError_t error_id = cudaGetDeviceCount(&deviceCount);

  if (error_id != cudaSuccess) {
    throw std::runtime_error("cudaGetDeviceCount failed: "s + std::to_string(error_id) +
                             ": "s + cudaGetErrorString(error_id));
  }

  for (int dev = 0; dev < deviceCount; ++dev) {
    cudaSetDevice(dev);
    cudaDeviceProp deviceProp;
    cudaGetDeviceProperties(&deviceProp, dev);

    if (deviceProp.major <= 0) {
      throw std::runtime_error("unexpected cuda compute capability: major "s +
                               std::to_string(deviceProp.major));
    }
    if (deviceProp.minor < 0) {
      throw std::runtime_error("unexpected cuda compute capability: minor "s +
                               std::to_string(deviceProp.minor));
    }

    if (deviceProp.major >= 10) {
      throw std::runtime_error("unexpected cuda compute capability: major "s +
                               std::to_string(deviceProp.major));
    }
    if (deviceProp.minor >= 10) {
      throw std::runtime_error("unexpected cuda compute capability: minor "s +
                               std::to_string(deviceProp.minor));
    }

    ret.push_back((deviceProp.major * 10U) + deviceProp.minor);
  }

  return ret;
}

Here is the call graph for this function:

Here is the caller graph for this function: