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

HeterogeneousCore/CUDAUtilities/interface/deviceCount.h

18 строк415 B

Andrea Bocci

Rename GPU check macros to uppercase

24 июл 2026, 18:36

24 июл 2026, 18:364ca8016

#ifndef HeterogenousCore_CUDAUtilities_deviceCount_h
#define HeterogenousCore_CUDAUtilities_deviceCount_h

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

#include <cuda_runtime.h>

namespace cms {
  namespace cuda {
    inline int deviceCount() {
      int ndevices;
      CUDA_CHECK(cudaGetDeviceCount(&ndevices));
      return ndevices;
    }
  }  // namespace cuda
}  // namespace cms

#endif