Hacker News · 首页· alxmrs·· 3 小时前AI 评分38
Show HN:从约 25 行 SQL 自动推导 FlashAttention kernel
Show HN: Automatic deduction of FlashAttention kernel from ~25 lines of SQL
AI 导读
一个 Show HN 项目展示了从约 25 行 SQL 自动推导出 FlashAttention kernel 的能力,相关实现位于 xarray-sql 仓库的 benchmarks/ml 目录下,对应文件为 01_attention.py(322 行、11.7 KB)。
来源:Hacker News · 首页 · github.com