Commits · 6ac8e63a85acdf717783873c67eb4cb648c621a4 · gaoqiong / flash-attention

28 Mar, 2024 1 commit
- flash_attn -> vllm_flash_attn · 6ac8e63a
  Woosuk Kwon authored Mar 28, 2024
  
  6ac8e63a
04 Sep, 2023 1 commit
- [MLP] Implement SwiGLU with torch jiterator · 3557e0bb
  Tri Dao authored Sep 04, 2023
  
  3557e0bb
18 Aug, 2023 1 commit
- Run isort and black on python files · f1a73d07
  Tri Dao authored Aug 18, 2023
  
  f1a73d07
13 Apr, 2023 1 commit
- [FusedDense] Enable sqrelu activation in FusedMLP · 6f6e9a9a
  Tri Dao authored Apr 13, 2023
  
  6f6e9a9a
14 Nov, 2022 1 commit
- Add fused_dense and dropout_add_layernorm CUDA extensions · fa6d1ce4
  Tri Dao authored Nov 13, 2022
  
  fa6d1ce4