File size: 225 Bytes
490d9fe | 1 2 3 4 5 6 7 8 9 | """my_softmax_function — fused softmax-attention kernel (CUTLASS Python DSL).
Public API exposed to `kernels` consumers.
"""
from ._kernel import attention, softmax_attention
__all__ = ["attention", "softmax_attention"]
|