File size: 337 Bytes
4e689f6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
"""
Kernel acceleration module for Q-TensorFormer.
"""

from .tt_kernels import (
    slice_tt_cores,
    optimized_tt_contract,
    fast_tt_linear,
    benchmark_tt_kernel,
    TRITON_AVAILABLE,
)

__all__ = [
    "slice_tt_cores",
    "optimized_tt_contract",
    "fast_tt_linear",
    "benchmark_tt_kernel",
    "TRITON_AVAILABLE",
]