Commit Graph

3 Commits

Author SHA1 Message Date
oahzxl
501a9e9cd2 [hotfix] polish flash attention (#1802) 2022-11-07 14:30:22 +08:00
Jiarui Fang
c248800359 [kernel] skip tests of flash_attn and triton when they are not available (#1798) 2022-11-07 13:41:13 +08:00
oahzxl
25952b67d7 [feat] add flash attention (#1762) 2022-10-26 16:15:52 +08:00