Merge pull request #29061 from makortel/fixCUDAESProduct

Fix cms::cuda::ESProduct for zero devices

Author: cmsbuild

HeterogeneousCore/CUDACore/interface/ESProduct.h

@@ -7,9 +7,9 @@
 #include <vector>
 
 #include "FWCore/Utilities/interface/thread_safety_macros.h"
+#include "HeterogeneousCore/CUDAServices/interface/numberOfDevices.h"
 #include "HeterogeneousCore/CUDAUtilities/interface/EventCache.h"
 #include "HeterogeneousCore/CUDAUtilities/interface/cudaCheck.h"
-#include "HeterogeneousCore/CUDAUtilities/interface/deviceCount.h"
 #include "HeterogeneousCore/CUDAUtilities/interface/currentDevice.h"
 #include "HeterogeneousCore/CUDAUtilities/interface/eventWorkHasCompleted.h"
 
@@ -18,7 +18,7 @@ namespace cms {
     template <typename T>
     class ESProduct {
     public:
-      ESProduct() : gpuDataPerDevice_(deviceCount()) {
+      ESProduct() : gpuDataPerDevice_(numberOfDevices()) {
         for (size_t i = 0; i < gpuDataPerDevice_.size(); ++i) {
           gpuDataPerDevice_[i].m_event = getEventCache().get();
         }

HeterogeneousCore/CUDAServices/interface/numberOfCUDADevices.h

@@ -0,0 +1,14 @@
+#ifndef HeterogeneousCore_CUDAServices_numberOfDevices_h
+#define HeterogeneousCore_CUDAServices_numberOfDevices_h
+
+namespace cms {
+  namespace cuda {
+    // Returns the number of CUDA devices
+    // The difference wrt. the standard CUDA function or
+    // cms::cuda::deviceCount() is that if CUDAService is disabled,
+    // this function returns 0.
+    int numberOfDevices();
+  }  // namespace cuda
+}  // namespace cms
+
+#endif

HeterogeneousCore/CUDAServices/interface/numberOfDevices.h

@@ -0,0 +1,10 @@
+#include "HeterogeneousCore/CUDAServices/interface/numberOfDevices.h"
+#include "HeterogeneousCore/CUDAServices/interface/CUDAService.h"
+#include "FWCore/ServiceRegistry/interface/Service.h"
+
+namespace cms::cuda {
+  int numberOfDevices() {
+    edm::Service<CUDAService> cs;
+    return cs->enabled() ? cs->numberOfDevices() : 0;
+  }
+}  // namespace cms::cuda