Always add 'GPU' to ENABLE_BOT_TESTS; remove gpu types from ENABLE_BO…

…T_TESTS
cms-sw · Feb 26, 2025 · e7c5ac8 · e7c5ac8
1 parent 2ddc6ae
commit e7c5ac8
Showing 1 changed file with 20 additions and 6 deletions.
diff --git a/process_pr.py b/process_pr.py
@@ -143,12 +143,13 @@ def format(s, **kwds):
 REGEX_IGNORE_FILE_COUNT = r"\+file-count"
 TEST_WAIT_GAP = 720
 ALL_CHECK_FUNCTIONS = None
-GPU_FLAVORS = open(join(dirname(__file__), "gpu_flavors.txt"), "r").read().splitlines()
-EXTRA_RELVALS_TESTS = ["threading", "gpu", "high-stats", "nano"] + GPU_FLAVORS
+ALL_GPU_FLAVORS = open(join(dirname(__file__), "gpu_flavors.txt"), "r").read().splitlines()
+EXTRA_RELVALS_TESTS = ["threading", "gpu", "high-stats", "nano"]
 EXTRA_RELVALS_TESTS_OPTS = "_" + "|_".join(EXTRA_RELVALS_TESTS)
 EXTRA_TESTS = (
     "|".join(EXTRA_RELVALS_TESTS)
-    + "|hlt_p2_integration|hlt_p2_timing|profiling|none|multi-microarchs"
+    + "|hlt_p2_integration|hlt_p2_timing|profiling|none|multi-microarchs|"
+    + "|".join(ALL_GPU_FLAVORS)
 )
 SKIP_TESTS = "|".join(["static", "header"])
 ENABLE_TEST_PTRN = "enable(_test(s|)|)"
@@ -1593,19 +1594,32 @@ def process_pr(repo_config, gh, repo, issue, dryRun, cmsbuild_user=None, force=F
 
     # end of parsing comments section
 
-    # Extract enabled GPU flavors
-    for gpu_lc in GPU_FLAVORS:
+    # Extract enabled GPU flavors and remove them from global_test_params and enable_tests
+    for gpu_lc in ALL_GPU_FLAVORS:
         gpu_uc = gpu_lc.upper()
         if gpu_uc in enable_tests or gpu_uc in global_test_params.get("ENABLE_BOT_TESTS", ""):
             enabled_gpu_flavors.append(gpu_lc)
 
     if not enabled_gpu_flavors and (
         "GPU" in enable_tests or "GPU" in global_test_params.get("ENABLE_BOT_TESTS", "")
     ):
-        enabled_gpu_flavors = GPU_FLAVORS
+        enabled_gpu_flavors = ALL_GPU_FLAVORS
 
     if enabled_gpu_flavors:
         global_test_params["ENABLE_GPU_FLAVORS"] = " ".join(enabled_gpu_flavors)
+        for gpu_lc in enabled_gpu_flavors:
+            gpu_uc = gpu_lc.upper()
+            enable_tests = enable_tests.replace(gpu_uc, "")
+            if "ENABLE_BOT_TESTS" in global_test_params:
+                global_test_params["ENABLE_BOT_TESTS"] = global_test_params[
+                    "ENABLE_BOT_TESTS"
+                ].replace(gpu_uc, "")
+
+        enable_tests = re.sub(r"\s+", " ", enable_tests)
+        if "ENABLE_BOT_TESTS" in global_test_params:
+            global_test_params["ENABLE_BOT_TESTS"] = re.sub(
+                r"\s+", " ", global_test_params["ENABLE_BOT_TESTS"]
+            )
 
     # Check if it needs to be automatically closed.
     if mustClose: