hipsolver & rocsolver Test for the ROCK

.github/workflows/ci_linux.yml

@@ -83,4 +83,4 @@ jobs:
     with:
       artifact_run_id: "${{ inputs.artifact_run_id != '' && inputs.artifact_run_id || github.run_id }}"
       amdgpu_family: ${{ inputs.amdgpu_families }}
-      package_version: "7.0.0.dev0"  # TODO: setup version with a script/workflow
+      package_version: "7.0.0.dev0" # TODO: setup version with a script/workflow

build_tools/github_actions/fetch_test_configurations.py

@@ -54,6 +54,24 @@ def _get_script_path(script_name: str) -> str:
         "platform": ["linux", "windows"],
         "total_shards": 4,
     },
+    # SOLVER tests
+    "hipsolver": {
+        "job_name": "hipsolver",
+        "fetch_artifact_args": "--blas --tests",
+        "timeout_minutes": 5,
+        "test_script": f"python {_get_script_path('test_hipsolver.py')}",
+        "platform": ["linux", "windows"],
+        "total_shards": 1,
+    },
+    "rocsolver": {
+        "job_name": "rocsolver",
+        "fetch_artifact_args": "--blas --tests",
+        "timeout_minutes": 5,
+        "test_script": f"python {_get_script_path('test_rocsolver.py')}",
+        # Issue for adding windows tests: https://github.com/ROCm/TheRock/issues/1770
+        "platform": ["linux"],
+        "total_shards": 1,
+    },
     # PRIM tests
     "rocprim": {
         "job_name": "rocprim",

build_tools/github_actions/test_executable_scripts/test_hipsolver.py

@@ -0,0 +1,48 @@
+import logging
+import os
+import shlex
+import subprocess
+from pathlib import Path
+
+THEROCK_BIN_DIR = os.getenv("THEROCK_BIN_DIR")
+SCRIPT_DIR = Path(__file__).resolve().parent
+THEROCK_DIR = SCRIPT_DIR.parent.parent.parent
+
+PLATFORM = os.getenv("PLATFORM")
+AMDGPU_FAMILIES = os.getenv("AMDGPU_FAMILIES")
+
+# GTest sharding
+SHARD_INDEX = os.getenv("SHARD_INDEX", 1)
+TOTAL_SHARDS = os.getenv("TOTAL_SHARDS", 1)
+envion_vars = os.environ.copy()
+# For display purposes in the GitHub Action UI, the shard array is 1th indexed. However for shard indexes, we convert it to 0th index.
+envion_vars["GTEST_SHARD_INDEX"] = str(int(SHARD_INDEX) - 1)
+envion_vars["GTEST_TOTAL_SHARDS"] = str(TOTAL_SHARDS)
+
+logging.basicConfig(level=logging.INFO)
+
+tests_to_exclude = [
+    "*known_bug*",
+    "*HEEVD*float_complex*",
+    "*HEEVJ*float_complex*",
+    "*HEGVD*float_complex*",
+    "*HEGVJ*float_complex*",
+    "*HEEVDX*float_complex*",
+    "*SYTRF*float_complex*",
+    "*HEEVD*double_complex*",
+    "*HEEVJ*double_complex*",
+    "*HEGVD*double_complex*",
+    "*HEGVJ*double_complex*",
+    "*HEEVDX*double_complex*",
+    "*SYTRF*double_complex*",
+]
+
+exclusion_list = ":".join(tests_to_exclude)
+
+cmd = [
+    f"{THEROCK_BIN_DIR}/hipsolver-test",
+    f"--gtest_filter=-{exclusion_list}",
+]
+
+logging.info(f"++ Exec [{THEROCK_DIR}]$ {shlex.join(cmd)}")
+subprocess.run(cmd, cwd=THEROCK_DIR, check=True, env=envion_vars)

build_tools/github_actions/test_executable_scripts/test_rocsolver.py

@@ -0,0 +1,28 @@
+import logging
+import os
+import shlex
+import subprocess
+from pathlib import Path
+
+THEROCK_BIN_DIR = os.getenv("THEROCK_BIN_DIR")
+OUTPUT_ARTIFACTS_DIR = os.getenv("OUTPUT_ARTIFACTS_DIR")
+SCRIPT_DIR = Path(__file__).resolve().parent
+THEROCK_DIR = SCRIPT_DIR.parent.parent.parent
+
+logging.basicConfig(level=logging.INFO)
+
+# GTest sharding
+SHARD_INDEX = os.getenv("SHARD_INDEX", 1)
+TOTAL_SHARDS = os.getenv("TOTAL_SHARDS", 1)
+envion_vars = os.environ.copy()
+# For display purposes in the GitHub Action UI, the shard array is 1th indexed. However for shard indexes, we convert it to 0th index.
+envion_vars["GTEST_SHARD_INDEX"] = str(int(SHARD_INDEX) - 1)
+envion_vars["GTEST_TOTAL_SHARDS"] = str(TOTAL_SHARDS)
+
+cmd = [
+    f"{THEROCK_BIN_DIR}/rocsolver-test",
+    "--gtest_filter=*known_bug*",
+]
+
+logging.info(f"++ Exec [{THEROCK_DIR}]$ {shlex.join(cmd)}")
+subprocess.run(cmd, cwd=THEROCK_DIR, check=True, env=envion_vars)