Enable ocltst execution on Rock

build_tools/github_actions/fetch_test_configurations.py

@@ -71,6 +71,20 @@ def _get_script_path(script_name: str) -> str:
             "windows": 4,
         },
     },
+    # ocltest
+    "ocltst": {
+        "job_name": "ocltst",
+        "fetch_artifact_args": "--tests",
+        "timeout_minutes": 30,
+        "test_script": f"python {_get_script_path('test_ocltst.py')}",
+        "platform": ["linux", "windows"],
+        "total_shards_dict": {
+            "linux": 1,
+            "windows": 1,
+        },
+        "container_image": "ghcr.io/rocm/no_rocm_image_ubuntu24_04_ocl_rt@sha256:f059191fdd5cf3bf8c5a47cb017a7be6e9bded0d52be594feb96f724f6a838eb",
+        "container_options": "--cap-add=SYS_PTRACE --cap-add=SYS_NICE",
+    },
     # BLAS tests
     "rocblas": {
         "job_name": "rocblas",

build_tools/github_actions/test_executable_scripts/test_ocltst.py

@@ -0,0 +1,96 @@
+import logging
+import os
+import shlex
+import subprocess
+from pathlib import Path
+import sys
+import platform
+import shutil
+
+logging.basicConfig(level=logging.INFO)
+THEROCK_BIN_DIR_STR = os.getenv("THEROCK_BIN_DIR")
+if THEROCK_BIN_DIR_STR is None:
+    logging.info(
+        "++ Error: env(THEROCK_BIN_DIR) is not set. Please set it before executing tests."
+    )
+    sys.exit(1)
+THEROCK_BIN_DIR = Path(THEROCK_BIN_DIR_STR).resolve()
+THEROCK_DIR = Path(THEROCK_BIN_DIR).parent
+env = os.environ.copy()
+is_windows = platform.system() == "Windows"
+
+
+# copies the dlls to local ocltst path.
+# to overwrite the registry entries
+def copy_dlls_exe_path(ocltst_path):
+    if platform.system() == "Windows":
+        # hip and comgr dlls need to be copied to the same folder as exectuable
+        dlls_pattern = ["amdocl*.dll", "amd_comgr*.dll", "OpenCL.dll"]
+        dlls_to_copy = []
+        for pattern in dlls_pattern:
+            dlls_to_copy.extend(THEROCK_BIN_DIR.glob(pattern))
+        for dll in dlls_to_copy:
+            try:
+                shutil.copy(dll, ocltst_path)
+                logging.info(f"++ Copied: {dll} to {ocltst_path}")
+            except Exception as e:
+                logging.info(f"++ Error copying {dll}: {e}")
+
+
+# returns ocltst path
+def setup_env(env):
+    ROCM_PATH = Path(THEROCK_DIR)
+    env["ROCM_PATH"] = str(ROCM_PATH)
+    if not is_windows:
+        ROCK_LIB_PATH = Path(THEROCK_DIR) / "lib"
+        OCL_LIB = Path(ROCK_LIB_PATH) / "opencl"
+        LLVM_LIB = Path(ROCK_LIB_PATH) / "llvm" / "lib"
+        ROCM_SYSDEPS_LIB = Path(ROCK_LIB_PATH) / "rocm_sysdeps" / "lib"
+        OCL_ICD_VENDORS = Path(THEROCK_DIR) / "etc" / "OpenCL" / "vendors"
+        OCLTST_PATH = Path(THEROCK_DIR) / "share" / "opencl" / "ocltst"
+        LD_LIBRARY_PATH = os.getenv("LD_LIBRARY_PATH")
+        if LD_LIBRARY_PATH is not None:
+            LD_LIBRARY_PATH = Path(LD_LIBRARY_PATH)
+        env["LD_LIBRARY_PATH"] = (
+            f"{ROCK_LIB_PATH}:{OCL_LIB}:{LLVM_LIB}:{ROCM_SYSDEPS_LIB}:{LD_LIBRARY_PATH}:{OCLTST_PATH}"
+        )
+        env["OCL_ICD_VENDORS"] = f"{OCL_ICD_VENDORS}/"
+    else:
+
+        OCLTST_PATH = Path(THEROCK_DIR) / "tests" / "ocltst"
+        copy_dlls_exe_path(OCLTST_PATH)
+        OCL_DLL_FILE = Path(OCLTST_PATH) / "amdocl64.dll"
+        OCL_ICD_DLL = Path(THEROCK_BIN_DIR) / "OpenCL.dll"
+        env["OCL_ICD_FILENAMES"] = str(OCL_DLL_FILE)
+    return OCLTST_PATH
+
+
+def execute_tests(env):
+    OCLTST_PATH = setup_env(env)
+    if not is_windows:
+        cmd = [
+            "./ocltst",
+            "-m",
+            "liboclruntime.so",
+            "-A",
+            "oclruntime.exclude",
+        ]
+        shell_var = False
+        logging.info(f"++ Setting LD_LIBRARY_PATH={env['LD_LIBRARY_PATH']}")
+        logging.info(f"++ Setting OCL_ICD_VENDORS={env['OCL_ICD_VENDORS']}")
+    else:
+        cmd = [
+            "ocltst.exe",
+            "-m",
+            "oclruntime.dll",
+            "-A",
+            "oclruntime.exclude",
+        ]
+        shell_var = True
+        logging.info(f"++ Setting OCL_ICD_FILENAMES={env['OCL_ICD_FILENAMES']}")
+    logging.info(f"++ Exec [{OCLTST_PATH}]$ {shlex.join(cmd)}")
+    subprocess.run(cmd, cwd=OCLTST_PATH, check=True, env=env, shell=shell_var)
+
+
+if __name__ == "__main__":
+    execute_tests(env)

core/CMakeLists.txt

@@ -374,6 +374,7 @@ if(THEROCK_ENABLE_OCL_RUNTIME)
 
   set(OCL_CLR_CMAKE_ARGS)
   set(OCL_CLR_RUNTIME_DEPS)
+
   if(WIN32)
     # Windows CLR options
     set(_compute_pal_dir "${THEROCK_ROCM_SYSTEMS_SOURCE_DIR}/shared/amdgpu-windows-interop")
@@ -394,10 +395,11 @@ if(THEROCK_ENABLE_OCL_RUNTIME)
       rocprofiler-register
       ROCR-Runtime
     )
-    list(APPEND OCL_CLR_CMAKE_ARGS
-      "-DBUILD_TESTS=${THEROCK_BUILD_TESTING}"
-    )
   endif()
+  # ocltst build flag
+  list(APPEND OCL_CLR_CMAKE_ARGS
+  "-DBUILD_TESTS=${THEROCK_BUILD_TESTING}"
+  )
 
   therock_cmake_subproject_declare(ocl-clr
     USE_DIST_AMDGPU_TARGETS