跳到正文
Hacker News· alxmrs·· 4 小时前AI 评分38

Show HN:从约 25 行 SQL 自动推导 FlashAttention kernel

Show HN: Automatic deduction of FlashAttention kernel from ~25 lines of SQL

AI 导读

有开发者展示了从约 25 行 SQL 自动推导出 FlashAttention kernel 的方法,相关代码位于 xarray-sql 项目的 benchmarks/ml 目录下,文件为 01_attention.py,共 322 行。

来源:Hacker News · github.com