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 的实现入口，便于对照研究注意力计算与代码生成方式。
正文 · 原文
Breadcrumbs
Latest commit
History
322 lines (282 loc) · 11.7 KB
Breadcrumbs
01_attention.py
File metadata and controls
322 lines (282 loc) · 11.7 KB
来源：Hacker News · github.com