跳到正文
Hacker News-首页· alxmrs·· 3 小时前AI 评分24

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 的方法。相关代码与基准测试内容已在开源仓库中更新。

正文 · 原文

Breadcrumbs

  1. xarray-sql
  2. benchmarks
  3. ml

Latest commit

History

History

322 lines (282 loc) · 11.7 KB

Breadcrumbs

  1. xarray-sql
  2. benchmarks
  3. ml

01_attention.py

File metadata and controls

322 lines (282 loc) · 11.7 KB

Raw

来源:Hacker News-首页 · github.com