"""kernbench.tp — Megatron-style Tensor Parallelism (ADR-0027).

Public API re-exports.
"""
from kernbench.tp.layers import (
    ColumnParallelLinear,
    RowParallelLinear,
)
from kernbench.tp.parallel_state import (
    get_tensor_model_parallel_rank,
    get_tensor_model_parallel_world_size,
    initialize_model_parallel,
)

__all__ = [
    "ColumnParallelLinear",
    "RowParallelLinear",
    "get_tensor_model_parallel_rank",
    "get_tensor_model_parallel_world_size",
    "initialize_model_parallel",
]