update runTests.py to make use of IB_TEST_TYPE if set

Author: smuzaffar

runTests.py

@@ -117,12 +117,20 @@ def run(self):
             return
         precmd = ""
         paralleJobs = MachineCPUCount
+        user_tests = os.environ.get("IB_TEST_TYPE", "")
+        if self.xType == "GPU":
+            user_tests = "cuda"
+        elif self.xType == "ROCM":
+            user_tests = "rocm"
         if ("ASAN" in os.environ["CMSSW_VERSION"]) or ("UBSAN" in os.environ["CMSSW_VERSION"]):
             paralleJobs = int(MachineCPUCount / 2)
-        if (self.xType == "GPU") or ("_GPU_X" in os.environ["CMSSW_VERSION"]):
-            precmd = "export USER_UNIT_TESTS=cuda ;"
-        if (self.xType == "ROCM") or ("_ROCM_X" in os.environ["CMSSW_VERSION"]):
-            precmd = "export USER_UNIT_TESTS=rocm ;"
+        if user_tests == "":
+            if "_GPU_X" in os.environ["CMSSW_VERSION"]:
+                user_tests = "cuda"
+            elif "_ROCM_X" in os.environ["CMSSW_VERSION"]:
+                user_tests = "rocm"
+        if user_tests != "":
+            precmd = "export USER_UNIT_TESTS=%s ;" % user_tests
         skiptests = ""
         if "lxplus" in getHostName():
             skiptests = "SKIP_UNITTESTS=ExpressionEvaluatorUnitTest"
@@ -432,6 +440,10 @@ def doTest(self, only=None):
             print("\n" + 80 * "-" + " gpu_unit \n")
             self.threadList["gpu_unit"] = self.runUnitTests([], "GPU")
 
+        if only and "rocm_unit" in only:
+            print("\n" + 80 * "-" + " rocm_unit \n")
+            self.threadList["rocm_unit"] = self.runUnitTests([], "ROCM")
+
         if not only or "codeRules" in only:
             print("\n" + 80 * "-" + " codeRules \n")
             self.threadList["codeRules"] = self.runCodeRulesChecker()