Dosyalar
rocm-systems/projects/rocprofiler-systems/tests/pytest/rocprofsys/__init__.py
T
Kian Cossettini 0eac446cb0 [rocprofiler-systems] - Implement subset of CTests into PyTests (#2666)
Convert a subset of the ctest to pytest to be used in TheRock CI.
Create a new cmake flag `ROCPROFSYS_INSTALL_TESTING` to control test suite installation.
- pytest package will be installed to share/rocprofiler-systems/tests
- all compiled examples are put in share/rocprofiler-systems/examples
- all test relevant scripts are put in share/rocprofiler-systems/tests
- see README.md in share/rocprofiler-systems/tests
2026-01-26 23:10:01 -05:00

75 satır
1.6 KiB
Python

# Copyright (c) Advanced Micro Devices, Inc.
# SPDX-License-Identifier: MIT
"""
rocprofsys testing utilities package.
Provides reusable components for testing rocprofiler-systems:
- Test runners (sampling, binary rewrite, runtime instrument)
- Output validators (perfetto, rocpd, timemory, regex patterns)
- Configuration management
- GPU and system detection utilities
"""
from .config import (
RocprofsysConfig,
discover_install_config,
discover_build_config,
)
from .runners import (
TestResult,
BaselineRunner,
SamplingRunner,
BinaryRewriteRunner,
RuntimeInstrumentRunner,
SysRunRunner,
)
from .validators import (
ValidationResult,
validate_perfetto_trace,
validate_rocpd_database,
validate_timemory_json,
validate_causal_json,
validate_file_exists,
validate_regex,
)
from .gpu import (
GPUInfo,
get_rocminfo,
detect_gpu,
lookup_gpu_category,
get_target_gpu_arch,
get_offload_extractor,
)
__all__ = [
# Config
"RocprofsysConfig",
"discover_build_config",
"discover_install_config",
# Runners
"TestResult",
"BaselineRunner",
"SamplingRunner",
"BinaryRewriteRunner",
"RuntimeInstrumentRunner",
"SysRunRunner",
# Validators
"ValidationResult",
"validate_perfetto_trace",
"validate_rocpd_database",
"validate_timemory_json",
"validate_causal_json",
"validate_file_exists",
"validate_regex",
# GPU
"GPUInfo",
"get_rocminfo",
"detect_gpu",
"lookup_gpu_category",
"get_target_gpu_arch",
"get_offload_extractor",
]