Commits · 50896ec57424f3ebe7977c0f10067887ec40b627 · gaoqiong / flash-attention

14 Mar, 2024 1 commit
- Make nvcc threads configurable via environment variable (#885) · 50896ec5
  Chirag Jain authored Mar 14, 2024
  
  50896ec5
15 Mar, 2023 1 commit
- Support H100 for other CUDA extensions · dc08ea1c
  Tri Dao authored Mar 15, 2023
  
  dc08ea1c
14 Nov, 2022 1 commit
- Add fused_dense and dropout_add_layernorm CUDA extensions · fa6d1ce4
  Tri Dao authored Nov 13, 2022
  
  fa6d1ce4
24 Oct, 2022 1 commit
- Add Megatron attention implementation for benchmarking · ed553e92
  Tri Dao authored Oct 23, 2022
  
  ed553e92