Add miopen-plugin project to TheRock

CMakeLists.txt

@@ -333,6 +333,12 @@ if(NOT WIN32)
     DESCRIPTION "Enables the hipDNN project"
     REQUIRES COMPILER HIP_RUNTIME
   )
+
+  therock_add_feature(MIOPEN_PLUGIN
+    GROUP ML_LIBS
+    DESCRIPTION "Enables the MIOpen plugin project"
+    REQUIRES COMPILER MIOPEN HIPDNN HIP_RUNTIME
+  )
 endif()
 
 # Finalize all feature flags.

README.md

@@ -158,22 +158,23 @@ Individual features can be controlled separately (typically in combination with
 `-DTHEROCK_ENABLE_ALL=OFF` or `-DTHEROCK_RESET_FEATURES=ON` to force a
 minimal build):
 
-| Component flag                     | Description                                   |
-| ---------------------------------- | --------------------------------------------- |
-| `-DTHEROCK_ENABLE_COMPILER=ON`     | Enables the GPU+host compiler toolchain       |
-| `-DTHEROCK_ENABLE_HIPIFY=ON`       | Enables the hipify tool                       |
-| `-DTHEROCK_ENABLE_CORE_RUNTIME=ON` | Enables the core runtime components and tools |
-| `-DTHEROCK_ENABLE_HIP_RUNTIME=ON`  | Enables the HIP runtime components            |
-| `-DTHEROCK_ENABLE_OCL_RUNTIME=ON`  | Enables the OpenCL runtime components         |
-| `-DTHEROCK_ENABLE_ROCPROFV3=ON`    | Enables rocprofv3                             |
-| `-DTHEROCK_ENABLE_RCCL=ON`         | Enables RCCL                                  |
-| `-DTHEROCK_ENABLE_PRIM=ON`         | Enables the PRIM library                      |
-| `-DTHEROCK_ENABLE_BLAS=ON`         | Enables the BLAS libraries                    |
-| `-DTHEROCK_ENABLE_RAND=ON`         | Enables the RAND libraries                    |
-| `-DTHEROCK_ENABLE_SOLVER=ON`       | Enables the SOLVER libraries                  |
-| `-DTHEROCK_ENABLE_SPARSE=ON`       | Enables the SPARSE libraries                  |
-| `-DTHEROCK_ENABLE_MIOPEN=ON`       | Enables MIOpen                                |
-| `-DTHEROCK_ENABLE_HIPDNN=ON`       | Enables hipDNN                                |
+| Component flag                      | Description                                   |
+| ----------------------------------- | --------------------------------------------- |
+| `-DTHEROCK_ENABLE_COMPILER=ON`      | Enables the GPU+host compiler toolchain       |
+| `-DTHEROCK_ENABLE_HIPIFY=ON`        | Enables the hipify tool                       |
+| `-DTHEROCK_ENABLE_CORE_RUNTIME=ON`  | Enables the core runtime components and tools |
+| `-DTHEROCK_ENABLE_HIP_RUNTIME=ON`   | Enables the HIP runtime components            |
+| `-DTHEROCK_ENABLE_OCL_RUNTIME=ON`   | Enables the OpenCL runtime components         |
+| `-DTHEROCK_ENABLE_ROCPROFV3=ON`     | Enables rocprofv3                             |
+| `-DTHEROCK_ENABLE_RCCL=ON`          | Enables RCCL                                  |
+| `-DTHEROCK_ENABLE_PRIM=ON`          | Enables the PRIM library                      |
+| `-DTHEROCK_ENABLE_BLAS=ON`          | Enables the BLAS libraries                    |
+| `-DTHEROCK_ENABLE_RAND=ON`          | Enables the RAND libraries                    |
+| `-DTHEROCK_ENABLE_SOLVER=ON`        | Enables the SOLVER libraries                  |
+| `-DTHEROCK_ENABLE_SPARSE=ON`        | Enables the SPARSE libraries                  |
+| `-DTHEROCK_ENABLE_MIOPEN=ON`        | Enables MIOpen                                |
+| `-DTHEROCK_ENABLE_MIOPEN_PLUGIN=ON` | Enables MIOpen_plugin                         |
+| `-DTHEROCK_ENABLE_HIPDNN=ON`        | Enables hipDNN                                |
 
 > [!TIP]
 > Enabling any features will implicitly enable their *minimum* dependencies. Some

build_tools/github_actions/fetch_test_configurations.py

@@ -187,6 +187,15 @@ def _get_script_path(script_name: str) -> str:
         "platform": ["linux"],
         "total_shards": 1,
     },
+    # MIOpen plugin tests
+    "miopen_plugin": {
+        "job_name": "miopen_plugin",
+        "fetch_artifact_args": "--blas --miopen --hipdnn --miopen-plugin --tests",
+        "timeout_minutes": 15,
+        "test_script": f"python {_get_script_path('test_miopen_plugin.py')}",
+        "platform": ["linux"],
+        "total_shards": 1,
+    },
 }
 
 

build_tools/github_actions/test_executable_scripts/test_hipdnn.py

@@ -18,7 +18,7 @@
     "--parallel",
     "8",
     "--timeout",
-    "10",
+    "60",
 ]
 logging.info(f"++ Exec [{THEROCK_DIR}]$ {shlex.join(cmd)}")
 

build_tools/github_actions/test_executable_scripts/test_miopen_plugin.py

@@ -0,0 +1,39 @@
+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
+
+logging.basicConfig(level=logging.INFO)
+
+cmd = [
+    "ctest",
+    "--test-dir",
+    f"{THEROCK_BIN_DIR}/miopen_legacy_plugin",
+    "--output-on-failure",
+    "--parallel",
+    "8",
+    "--timeout",
+    "600",
+]
+
+# Determine test filter based on TEST_TYPE environment variable
+environ_vars = os.environ.copy()
+test_type = os.getenv("TEST_TYPE", "full")
+
+if test_type == "smoke":
+    # Exclude tests that start with "Full" during smoke tests
+    environ_vars["GTEST_FILTER"] = "-Full*"
+
+logging.info(f"++ Exec [{THEROCK_DIR}]$ {shlex.join(cmd)}")
+
+subprocess.run(
+    cmd,
+    cwd=THEROCK_DIR,
+    check=True,
+    env=environ_vars,
+)

build_tools/install_rocm_from_artifacts.py

@@ -14,8 +14,8 @@
     (--artifact-group ARTIFACT_GROUP | --amdgpu_family AMDGPU_FAMILY)
     [--output-dir OUTPUT_DIR]
     (--run-id RUN_ID | --release RELEASE | --input-dir INPUT_DIR)
-    [--blas | --no-blas] [--fft | --no-fft] [--hipdnn | --no-hipdnn] [--miopen | --no-miopen] [--prim | --no-prim]
-    [--rand | --no-rand] [--rccl | --no-rccl] [--tests | --no-tests] [--base-only]
+    [--blas | --no-blas] [--fft | --no-fft] [--hipdnn | --no-hipdnn] [--miopen | --no-miopen] [--miopen-plugin | --no-miopen-plugin]
+    [--prim | --no-prim] [--rand | --no-rand] [--rccl | --no-rccl] [--tests | --no-tests] [--base-only]
 
 Examples:
 - Downloads and unpacks the gfx94X S3 artifacts from GitHub CI workflow run 14474448215
@@ -161,7 +161,16 @@ def retrieve_artifacts_by_run_id(args):
     if args.base_only:
         argv.extend(base_artifact_patterns)
     elif any(
-        [args.blas, args.fft, args.hipdnn, args.miopen, args.prim, args.rand, args.rccl]
+        [
+            args.blas,
+            args.fft,
+            args.hipdnn,
+            args.miopen,
+            args.miopen_plugin,
+            args.prim,
+            args.rand,
+            args.rccl,
+        ]
     ):
         argv.extend(base_artifact_patterns)
 
@@ -175,6 +184,8 @@ def retrieve_artifacts_by_run_id(args):
             extra_artifacts.append("hipdnn")
         if args.miopen:
             extra_artifacts.append("miopen")
+        if args.miopen_plugin:
+            extra_artifacts.append("miopen-plugin")
         if args.prim:
             extra_artifacts.append("prim")
         if args.rand:
@@ -334,6 +345,13 @@ def main(argv):
         action=argparse.BooleanOptionalAction,
     )
 
+    artifacts_group.add_argument(
+        "--miopen-plugin",
+        default=False,
+        help="Include 'miopen-plugin' artifacts",
+        action=argparse.BooleanOptionalAction,
+    )
+
     artifacts_group.add_argument(
         "--prim",
         default=False,