| """ | |
| 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", | |
| ] | |