"""
FP16 性能评估 Benchmark 框架

用于评估 RTX 3050 GPU 在 YOLOv11n 模型下的 FP16 推理性能极限。
"""

__version__ = "1.0.0"
__author__ = "Edge AI System Engineer"

from .config import BenchmarkConfig, load_config, save_config
from .engine_builder import TRTEngineBuilder, EngineMetadata
from .inference_engine import TRTInferenceEngine
from .decode_thread import DecodeThread, FrameQueueManager
from .batch_assembler import BatchAssembler, GPUPreprocessor
from .metrics_collector import MetricsCollector, GPUMonitor
from .results import TestResult, export_json, export_csv, generate_report
from .benchmark_runner import BenchmarkRunner
from .stress_test import StressTestRunner, run_stress_test

__all__ = [
    "BenchmarkConfig",
    "load_config",
    "save_config",
    "TRTEngineBuilder",
    "EngineMetadata",
    "TRTInferenceEngine",
    "DecodeThread",
    "FrameQueueManager",
    "BatchAssembler",
    "GPUPreprocessor",
    "MetricsCollector",
    "GPUMonitor",
    "TestResult",
    "export_json",
    "export_csv",
    "generate_report",
    "BenchmarkRunner",
    "StressTestRunner",
    "run_stress_test",
]