跳到正文
原文
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