"""
TileLang-accelerated SSM operations.

This module provides GPU-accelerated selective scan operations using TileLang principles,
with Triton optimization and PyTorch fallback.
"""

from .selective_scan import (
    HAS_TILELANG,
    HAS_TILELANG_ACCELERATION,
    HAS_TRITON,
    TILELANG_BACKEND,
    ssm_gamma_forward_tilelang,
)

__all__ = [
    "ssm_gamma_forward_tilelang",
    "HAS_TRITON",
    "HAS_TILELANG",
    "HAS_TILELANG_ACCELERATION",
    "TILELANG_BACKEND",
]