跳到正文
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 的实现入口,便于对照研究注意力计算与代码生成方式。

正文 · AI 翻译

面包屑导航

  1. xarray-sql
  2. 基准测试
  3. ml

最新提交

历史记录

历史记录

322 行 (282 loc) · 11.7 KB

面包屑导航

  1. xarray-sql
  2. 基准测试
  3. ml

01_attention.py

文件元数据和控制项

322 行 (282 loc) · 11.7 KB

原始内容

来源:Hacker News · github.com