"""Benchmark utilities for sagellm-comm.

This module provides micro-benchmarks for collective operations.
"""

from __future__ import annotations

__all__ = ["CollectiveBenchmark", "BenchmarkResult", "run_all_reduce_benchmark"]

from sagellm_comm.benchmark.collective import (
    BenchmarkResult,
    CollectiveBenchmark,
    run_all_reduce_benchmark,
)
