Commits

Branch selector

sageattention-1

User selector

All users

Datepicker

All time

Commit history

Commits on Aug 5, 2025

• Merge pull request #230 from BobQC/sageattention-1

  Show description for 2bec28a
  whx1003

  authored
  2bec28aView commit details

  Copy full SHA for 2bec28a

  Browse repository at this point

• fix bug: no skip when mask_dtype is float

  BobQC

  committed
  bea7ba8View commit details

  Copy full SHA for bea7ba8

  Browse repository at this point

• Merge pull request #226 from BobQC/sageattention-1

  Show description for 2be74af
  whx1003

  authored
  2be74afView commit details

  Copy full SHA for 2be74af

  Browse repository at this point

• fix qk_mask when mask_dtype=float

  BobQC

  committed
  c4d922dView commit details

  Copy full SHA for c4d922d

  Browse repository at this point

Commits on Aug 4, 2025

• fix qk_mask

  BobQC

  committed
  13b7cd3View commit details

  Copy full SHA for 13b7cd3

  Browse repository at this point

• Merge pull request #225 from thu-ml/fix/attention-mask

  Show description for 66ab6e9
  whx1003

  authored
  66ab6e9View commit details

  Copy full SHA for 66ab6e9

  Browse repository at this point

• fix mask computation for better precision

  whx1003

  committed
  acd0939View commit details

  Copy full SHA for acd0939

  Browse repository at this point

Commits on May 2, 2025

• Update Citation

  jt-zhang

  authored
  19d3288View commit details

  Copy full SHA for 19d3288

  Browse repository at this point

• Update README.md

  jt-zhang

  authored
  07b59afView commit details

  Copy full SHA for 07b59af

  Browse repository at this point

Commits on Dec 26, 2024

• rename resource to assets

  jt-zhang

  committed
  e4ac59aView commit details

  Copy full SHA for e4ac59a

  Browse repository at this point

Commits on Nov 21, 2024

• Update README

  jt-zhang

  committed
  3eaa8d1View commit details

  Copy full SHA for 3eaa8d1

  Browse repository at this point

Commits on Nov 20, 2024

• enable pip install sageattention==1.0.6

  jt-zhang

  committed
  0db4cbfView commit details

  Copy full SHA for 0db4cbf

  Browse repository at this point

• Update README.md

  jason-huang03

  authored
  b3eef25View commit details

  Copy full SHA for b3eef25

  Browse repository at this point

• add device and dtype check

  jason-huang03

  committed
  ee8a972View commit details

  Copy full SHA for ee8a972

  Browse repository at this point

Commits on Nov 19, 2024

• add sageattn2 news

  jt-zhang

  committed
  7b7b7e7View commit details

  Copy full SHA for 7b7b7e7

  Browse repository at this point

Commits on Nov 17, 2024

• fix typo

  jason-huang03

  committed
  52d2c4fView commit details

  Copy full SHA for 52d2c4f

  Browse repository at this point

Commits on Nov 15, 2024

• Update README.md

  jt-zhang

  authored
  14f75a2View commit details

  Copy full SHA for 14f75a2

  Browse repository at this point

Commits on Nov 12, 2024

• add pip install sageattention==1.0.5

  jt-zhang

  committed
  c883aaeView commit details

  Copy full SHA for c883aae

  Browse repository at this point

Commits on Nov 11, 2024

• v1.0.4 (#38)

  Show description for 3ce346d
  jason-huang03

  authored
  3ce346dView commit details

  Copy full SHA for 3ce346d

  Browse repository at this point

• add sageattn_varlen support (#36)

  Show description for 942850a
  jason-huang03

  authored
  942850aView commit details

  Copy full SHA for 942850a

  Browse repository at this point

• Merge pull request #35 from jason-huang03/main

  Show description for 33173b3
  jason-huang03

  authored
  33173b3View commit details

  Copy full SHA for 33173b3

  Browse repository at this point

• better support for hd96

  jason-huang03

  committed
  5ace8e2View commit details

  Copy full SHA for 5ace8e2

  Browse repository at this point

• Support different seqlen between `q` and `k,v`, (b,h,s,d) or (b,s,h,d) input, and group-query attention

  jt-zhang

  committed
  cacbefcView commit details

  Copy full SHA for cacbefc

  Browse repository at this point

• Merge pull request #33 from jason-huang03/main

  Show description for 3ca1381
  jt-zhang

  authored
  3ca1381View commit details

  Copy full SHA for 3ca1381

  Browse repository at this point

• clean this pr

  jt-zhang

  committed
  2b21d6cView commit details

  Copy full SHA for 2b21d6c

  Browse repository at this point

Commits on Nov 10, 2024

• fuse sub k mean into quant kernel

  jason-huang03

  committed
  ec342dbView commit details

  Copy full SHA for ec342db

  Browse repository at this point

• convert x to float32 inside quant kernel

  jason-huang03

  committed
  974c7a7View commit details

  Copy full SHA for 974c7a7

  Browse repository at this point

• [update] support non-contiguous input, HDN and HND layout, different kv_len and qo_len, GQA

  jason-huang03

  committed
  44b17d4View commit details

  Copy full SHA for 44b17d4

  Browse repository at this point

• add kernel for different qo_len, kv_len, GQA and non contiguous input

  jason-huang03

  committed
  873a32cView commit details

  Copy full SHA for 873a32c

  Browse repository at this point

Commits on Nov 3, 2024

• Update README.md

  jt-zhang

  authored
  63bd567View commit details

  Copy full SHA for 63bd567

  Browse repository at this point

Commits on Oct 24, 2024

• Update README.md

  jt-zhang

  authored
  1d8f531View commit details

  Copy full SHA for 1d8f531

  Browse repository at this point

• add seq_len and headdim assertion

  jt-zhang

  committed
  50af5d3View commit details

  Copy full SHA for 50af5d3

  Browse repository at this point

Commits on Oct 23, 2024

• Update README.md

  jt-zhang

  authored
  6e16c04View commit details

  Copy full SHA for 6e16c04

  Browse repository at this point

Commits on Oct 11, 2024

• Update README.md

  jt-zhang

  authored
  a46d1e3View commit details

  Copy full SHA for a46d1e3

  Browse repository at this point

Commits on Oct 10, 2024

• Update README.md

  jt-zhang

  authored
  bc0b3d4View commit details

  Copy full SHA for bc0b3d4

  Browse repository at this point