"""Thin re-export shim for the GEMM figure tests.

Not a test module (no ``test_`` prefix → pytest does not collect it).

The sweep + renderer logic now lives in
``kernbench.benches.milestone_1h_gemm`` (production single home, ADR-0054,
also driven by ``scripts/gemm_sweep.py``). The figure tests import the same
names from here; behavior is unchanged (defaults still target
``docs/diagrams/gemm_plots/``).
"""
from __future__ import annotations

from kernbench.benches.milestone_1h_gemm import (
    DEFAULT_PLOTS_DIR as GEMM_PLOTS_DIR,
    DEFAULT_SWEEP_JSON as GEMM_SWEEP_JSON,
    ROOT,
    emit_all_gemm_plots,
    emit_mac_utilization_measured,
    emit_mac_utilization_theoretical_vs_measured,
    emit_stage_breakdown,
)

__all__ = [
    "GEMM_PLOTS_DIR",
    "GEMM_SWEEP_JSON",
    "ROOT",
    "emit_all_gemm_plots",
    "emit_mac_utilization_measured",
    "emit_mac_utilization_theoretical_vs_measured",
    "emit_stage_breakdown",
]