"""Thin re-export shim for the GQA figure tests. Not a test module (no ``test_`` prefix → pytest does not collect it). Mirrors ``tests/gemm/_gemm_plot_helpers.py``. The renderer logic lives in ``kernbench.benches.milestone_gqa_llama70b`` (production single home, ADR-0054). Defaults still target the bench's ``_OUTPUT_DIR``. """ from __future__ import annotations from kernbench.benches.milestone_gqa_llama70b import ( _OUTPUT_DIR as GQA_PLOTS_DIR, _SWEEP_JSON as GQA_SWEEP_JSON, emit_all_gqa_plots, emit_gqa_comparison, emit_panel_op_log_summary, ) __all__ = [ "GQA_PLOTS_DIR", "GQA_SWEEP_JSON", "emit_all_gqa_plots", "emit_gqa_comparison", "emit_panel_op_log_summary", ]