Spaces:
Running on Zero
Running on Zero
File size: 337 Bytes
7975a75 | 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",
]
|