HeterogeneousCore/CUDAUtilities/interface/requireDevices.h - master - githubmirror/cmssw

HeterogeneousCore/CUDAUtilities/interface/requireDevices.h

17 строк537 B

Andrea Bocci

Import CUDA interaction layer from Patatrack, part 1

18 янв 2020, 01:00

18 янв 2020, 01:00f143441

#ifndef HeterogeneousCore_CUDAUtilities_requireDevices_h
#define HeterogeneousCore_CUDAUtilities_requireDevices_h

/**
 * These functions are meant to be called only from unit tests.
 */
namespace cms {
  namespace cudatest {
    /// In presence of CUDA devices, return true; otherwise print message and return false
    bool testDevices();

    /// Print message and exit if there are no CUDA devices
    void requireDevices();
  }  // namespace cudatest
}  // namespace cms

#endif  // HeterogeneousCore_CUDAUtilities_requireDevices_h