Hacker News· alxmrs·· 4 小时前精选AI 评分52
xarray-sql 展示从约 25 行 SQL 自动推导 FlashAttention kernel
Show HN: Automatic deduction of FlashAttention kernel from ~25 lines of SQL
AI 导读
xarray-sql 展示如何从约 25 行 SQL 自动推导 FlashAttention kernel。对应文件 `benchmarks/ml/01_attention.py` 共 322 行(282 loc),大小为 11.7 KB。
推荐理由
材料给出从约 25 行 SQL 自动推导 FlashAttention kernel 的实现入口,便于对照研究注意力计算与代码生成方式。
正文 · AI 翻译
面包屑导航
最新提交
历史记录
322 行 (282 loc) · 11.7 KB
面包屑导航
01_attention.py
文件元数据和控制项
322 行 (282 loc) · 11.7 KB
来源:Hacker News · github.com