Add 'CUDAService' category also when CUDA is disabled, but only if it…

… does not exist yet
cms-sw · Feb 24, 2022 · c02ab88 · c02ab88
1 parent 7d90aca
commit c02ab88
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/HeterogeneousCore/CUDACore/python/ProcessAcceleratorCUDA.py b/HeterogeneousCore/CUDACore/python/ProcessAcceleratorCUDA.py
@@ -19,9 +19,11 @@ def apply(self, process, accelerators):
             from HeterogeneousCore.CUDAServices.CUDAService_cfi import CUDAService
             process.add_(CUDAService)
 
+        if not hasattr(process.MessageLogger, "CUDAService"):
+            process.MessageLogger.CUDAService = cms.untracked.PSet()
+
         if self._label in accelerators:
             process.CUDAService.enabled = True
-            process.MessageLogger.CUDAService = cms.untracked.PSet()
         else:
             process.CUDAService.enabled = False