![]() |
|
|||
File indexing completed on 2024-04-06 12:15:44
0001 #ifndef HeterogenousCore_CUDAUtilities_deviceCount_h 0002 #define HeterogenousCore_CUDAUtilities_deviceCount_h 0003 0004 #include "HeterogeneousCore/CUDAUtilities/interface/cudaCheck.h" 0005 0006 #include <cuda_runtime.h> 0007 0008 namespace cms { 0009 namespace cuda { 0010 inline int deviceCount() { 0011 int ndevices; 0012 cudaCheck(cudaGetDeviceCount(&ndevices)); 0013 return ndevices; 0014 } 0015 } // namespace cuda 0016 } // namespace cms 0017 0018 #endif
[ Source navigation ] | [ Diff markup ] | [ Identifier search ] | [ general search ] |
This page was automatically generated by the 2.2.1 LXR engine. The LXR team |
![]() ![]() |