跳到正文
Hacker News 首页alxmrs

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

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

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

读原文 ↗导出 Markdown