Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • carlushuang's avatar
    support batch & nhead, and scale (#20) · 95889861
    carlushuang authored 
    * support batch & nhead

* support scale

* tile scheduler

* rename tile-scheduler to tile-partitioner

* add some exp2 math

* fix a bug when chaning tile size
    95889861
fmha_fwd_kernel.hpp 8 KB