inclusionAI 开源 cuLA:面向 Blackwell 与 Hopper 的线性注意力 CUDA 内核库
inclusionAI/cuLA
阅读原文
本站未展示全文,请前往来源网站阅读。
AI 导读
inclusionAI 开源 cuLA,用 CuTe DSL 和 CUTLASS C++ 为 GLA、KDA、GDN、Lightning Attention 等线性注意力变体提供手调 CUDA 内核,支持 NVIDIA Blackwell(SM10X)与 Hopper(SM90)GPU。
来源:inclusionAI GitHub 新仓库 · github.com