derek.holloway
7502bf64a4
CodeQL / Analyze (actions) (push) Canceled after 0s
CodeQL / Analyze (python) (push) Canceled after 0s
pre-commit / pre-commit (push) Canceled after 0s
Multi-Arch CI / setup (push) Canceled after 0s
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Setup (Multi-Arch) / setup (push) Canceled after 0s
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Multi-Arch CI / CI Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
79 lines
2.3 KiB
Python
79 lines
2.3 KiB
Python
# Copyright Advanced Micro Devices, Inc.
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
import logging
|
|
import os
|
|
import shlex
|
|
import subprocess
|
|
from pathlib import Path
|
|
|
|
THEROCK_BIN_DIR = os.getenv("THEROCK_BIN_DIR")
|
|
AMDGPU_FAMILIES = os.getenv("AMDGPU_FAMILIES")
|
|
platform = os.getenv("RUNNER_OS").lower()
|
|
SCRIPT_DIR = Path(__file__).resolve().parent
|
|
THEROCK_DIR = Path(
|
|
os.environ.get("THEROCK_DIR") or SCRIPT_DIR.parent.parent.parent
|
|
).resolve()
|
|
|
|
# GTest sharding
|
|
SHARD_INDEX = os.getenv("SHARD_INDEX", 1)
|
|
TOTAL_SHARDS = os.getenv("TOTAL_SHARDS", 1)
|
|
environ_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.
|
|
environ_vars["GTEST_SHARD_INDEX"] = str(int(SHARD_INDEX) - 1)
|
|
environ_vars["GTEST_TOTAL_SHARDS"] = str(TOTAL_SHARDS)
|
|
|
|
# Enable GTest "brief" output: only show failures and the final results
|
|
environ_vars["GTEST_BRIEF"] = str(1)
|
|
|
|
# Some of our runtime kernel compilations have been relying on either ROCM_PATH being set, or ROCm being installed at
|
|
# /opt/rocm. Neither of these is true in TheRock so we need to supply ROCM_PATH to our tests.
|
|
ROCM_PATH = Path(THEROCK_BIN_DIR).resolve().parent
|
|
environ_vars["ROCM_PATH"] = str(ROCM_PATH)
|
|
|
|
logging.basicConfig(level=logging.INFO)
|
|
|
|
# If quick tests are enabled, we run quick tests only.
|
|
# Otherwise, we run the normal test suite
|
|
test_type = os.getenv("TEST_TYPE", "full")
|
|
|
|
# TODO(#2823): Re-enable test once flaky issue is resolved
|
|
TESTS_TO_IGNORE = ["unpack_util_test"]
|
|
|
|
test_subdir = ""
|
|
timeout = "3600"
|
|
if test_type == "quick":
|
|
# The emulator regression tests are very fast.
|
|
# If we need something even faster we can use "/smoke" here.
|
|
test_subdir = "/regression"
|
|
timeout = "720"
|
|
elif test_type == "regression":
|
|
test_subdir = "/regression"
|
|
timeout = "720"
|
|
|
|
# Make per-device adjustments
|
|
ctest_parallelism = "2"
|
|
if AMDGPU_FAMILIES == "gfx1153":
|
|
ctest_parallelism = "1"
|
|
|
|
cmd = [
|
|
"ctest",
|
|
"--test-dir",
|
|
f"{THEROCK_BIN_DIR}/rocwmma{test_subdir}",
|
|
"--output-on-failure",
|
|
"--parallel",
|
|
ctest_parallelism,
|
|
"--timeout",
|
|
timeout,
|
|
"--exclude-regex",
|
|
"|".join(TESTS_TO_IGNORE),
|
|
]
|
|
logging.info(f"++ Exec [{THEROCK_DIR}]$ {shlex.join(cmd)}")
|
|
|
|
subprocess.run(
|
|
cmd,
|
|
cwd=THEROCK_DIR,
|
|
check=True,
|
|
env=environ_vars,
|
|
)
|