• 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.cpp 11.7 KB