HeterogeneousCore/CUDAUtilities/src/requireDevices.cc - master - githubmirror/cmssw

HeterogeneousCore/CUDAUtilities/src/requireDevices.cc

30 строк718 B

Andrea Bocci

Import CUDA interaction layer from Patatrack, part 1

18 янв 2020, 01:00

18 янв 2020, 01:00f143441

#include <cstdlib>
#include <iostream>

#include <cuda_runtime.h>

#include "HeterogeneousCore/CUDAUtilities/interface/requireDevices.h"

namespace cms::cudatest {
  bool testDevices() {
    int devices = 0;
    auto status = cudaGetDeviceCount(&devices);
    if (status != cudaSuccess) {
      std::cerr << "Failed to initialise the CUDA runtime, the test will be skipped."
                << "\n";
      return false;
    }
    if (devices == 0) {
      std::cerr << "No CUDA devices available, the test will be skipped."
                << "\n";
      return false;
    }
    return true;
  }

  void requireDevices() {
    if (not testDevices()) {
      exit(EXIT_SUCCESS);
    }
  }
}  // namespace cms::cudatest