跳到正文
Hacker News · 首页· alxmrs·· 16 天前AI 评分40

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