Safeguard in case of --gpu and no GPU available

Author: AdrianoDee

Configuration/PyReleaseValidation/python/relval_data_highstats.py

@@ -59,7 +59,7 @@
             wf_number = wf_number + offset_pd * p_n
             wf_number = wf_number + offset_events * evs
             wf_number = round(wf_number,6)
-            step_name = "Run" + pd + era.split("Run")[1] + "_10k"
+            step_name = "Run" + pd + era.split("Run")[1] + e_key
             y = str(int(base_wf))
             suff = "ZB_" if "ZeroBias" in step_name else ""
             workflows[wf_number] = ['',[step_name,'HLTDR3_' + y,'RECONANORUN3_' + suff + 'reHLT_'+y,'HARVESTRUN3_' + suff + y]]

Configuration/PyReleaseValidation/scripts/runTheMatrix.py

@@ -450,32 +450,42 @@ def runSelected(opt):
                           default='')
 
     opt = parser.parse_args()
-
     opt.selected_gpus = None
-    if opt.gpu:
+    
+    if opt.gpu != 'forbidden':
 
         print(">> Running with --gpu option. Checking the available and supported GPUs.")
         gpus = cleanComputeCapabilities("cuda")
         gpus = gpus + cleanComputeCapabilities("rocm", len(gpus))
         available_gpus = gpus
 
-        print("> GPUs Available:")
-        [print(f) for f in available_gpus]
+        if len(available_gpus) == 0:
+            print(">> No GPU available!")
+            opt.gpu = 'forbidden'
+        else:
+            print(">> GPUs available:")
+            [print(f) for f in available_gpus]
 
-        # Filtering ONLY CUDA GPUs on capability
-        gpus = [g for g in gpus if not g.isCUDA() or (g.isCUDA() and g.capability in opt.CUDACapabilities)]
+            # Filtering ONLY CUDA GPUs on capability
+            gpus = [g for g in gpus if not g.isCUDA() or (g.isCUDA() and g.capability in opt.CUDACapabilities)]
 
-        # Filtering by name (if parsed)
-        if len(opt.GPUName) > 0:
-            gpus = [g for g in gpus if g.name == opt.GPUName]
+            # Filtering by name (if parsed)
+            if len(opt.GPUName) > 0:
+                gpus = [g for g in gpus if g.name == opt.GPUName]
 
-        if available_gpus != gpus:
-            print(">> Selected:")   
-            [print(f) for f in gpus]
-        else:
-            print(">> All selected!")
-        
-        opt.selected_gpus = cycle(gpus)
+            if available_gpus != gpus:
+                if len(gpus) > 0:
+                    print(">> GPUs selected:")   
+                    [print(f) for f in gpus]
+                else:
+                    print(">> No GPU selected!")
+            else:
+                print(">> All selected!")
+
+            if len(gpus) > 0:
+                opt.selected_gpus = cycle(gpus)
+            else:
+                opt.gpu = 'forbidden'
 
     if opt.command: opt.command = ' '.join(opt.command)
     os.environ["CMSSW_DAS_QUERY_SITES"]=opt.dasSites