"""Summary attention kernels for KSA inference.""" from .interface import summary_attn_func