Commits · aa4fd2d16654a2105e92b761df73ed8222a098dd · Shenguo Wang / Flash Attention

05 Oct, 2023 1 commit
- Clarify that Windows is not supported right now · aa4fd2d1
  Tri Dao authored 1 year ago
  
  aa4fd2d1
04 Oct, 2023 2 commits
- [CI] Use official Pytorch 2.1, add CUDA 11.8 for Pytorch 2.1 · 5e525a8d
  Tri Dao authored 1 year ago
  
  5e525a8d
- Bump to v2.3.1 · 21c3b0d8
  Tri Dao authored 1 year ago
  
  21c3b0d8
03 Oct, 2023 1 commit
- [Gen] Accept cache_batch_idx to index into the KV cache · e279bf8e
  Tri Dao authored 1 year ago
  
  e279bf8e
27 Sep, 2023 1 commit
- Bump to v2.3.0 · 601b4dc4
  Tri Dao authored 1 year ago
  
  601b4dc4
26 Sep, 2023 1 commit
- Implement local attention · 083e8f52
  Tri Dao authored 1 year ago
```
Co-authored-by: Timothee Lacroix <t@mistral.ai>
```
  083e8f52
25 Sep, 2023 1 commit
- Fix NameError and typo in ApplyRotaryEmbQKV_ (#569) · 4c8ff915
  Katherine Crowson authored 1 year ago
  
  4c8ff915
24 Sep, 2023 4 commits
- Bump to v2.2.5 · 0a1d03c7
  Tri Dao authored 1 year ago
  
  0a1d03c7
- Switch cutlass to newer commit to avoid compilation warning · 812cb1c9
  Tri Dao authored 1 year ago
  
  812cb1c9
- Don't over-allocate dq_accum in case of varlen · 65c234ed
  Tri Dao authored 1 year ago
  
  65c234ed
- Reduce number of templates for headdim > 128 · 1879e089
  Tri Dao authored 1 year ago
  
  1879e089
22 Sep, 2023 2 commits
- Add placeholder for inference example · dd9a6fa4
  Tri Dao authored 1 year ago
  
  dd9a6fa4
- Re-enable compilation for Hopper · bff31471
  Tri Dao authored 1 year ago
  
  bff31471
21 Sep, 2023 4 commits
- Fix E1136 (#563) · 187c2a06
  Yuchao Dai authored 1 year ago
  
  187c2a06
- Bump to v2.2.4 · 229080b9
  Tri Dao authored 1 year ago
  
  229080b9
- Swap seqlen_q and ngroups when seqlen_q=1 (h/t Daniel Haziza) · 2d8ea9a5
  Tri Dao authored 1 year ago
  
  2d8ea9a5
- [Llama] Fix some tests, add tests for Llama 2 and CodeLlama · 0705d271
  Tri Dao authored 1 year ago
  
  0705d271
20 Sep, 2023 3 commits
- [Gen] Simplify decode_speculative · e0fbaa70
  Tri Dao authored 1 year ago
  
  e0fbaa70
- [Gen] Rename max_sequence_len->max_seqlen, sequence_len_offset->seqlen_offset · e6a80264
  Tri Dao authored 1 year ago
  
  e6a80264
- Fix Llama GQA/MQA (#546) · 42832575
  Kevin Hu authored 1 year ago
```
* Fix llama MQA

* Fix permute shape

* Update llama.py
```
  42832575
18 Sep, 2023 5 commits
- [Gen] Don't use ft_attention, use flash_attn_with_kvcache instead · dfe29f5e
  Tri Dao authored 1 year ago
  
  dfe29f5e
- Swap seqlen_q, nheads for MQA when seqlen_q=1 for fwd (h/t Daniel H) · 3250ff3d
  Tri Dao authored 1 year ago
  
  3250ff3d
- Remove template for (IsEvenMN=T, IsEvenK=F) to speed up compilation · 43617dea
  Tri Dao authored 1 year ago
  
  43617dea
- [Minor] add nvcc note on bare_metal_version `RuntimeError` (#552) · fa3ddcba
  Federico Berto authored 1 year ago
```
* Add nvcc note on bare_metal_version `RuntimeError`

* Run Black formatting
```
  fa3ddcba
- Don't compile for Pytorch 2.1 on CUDA 12.1 due to nvcc segfaults · 799f56fa
  Tri Dao authored 1 year ago
  
  799f56fa
17 Sep, 2023 1 commit
- Set block size to 64 x 64 for kvcache to avoid nvcc segfaults · c984208d
  Tri Dao authored 1 year ago
  
  c984208d
16 Sep, 2023 3 commits
- Bump to v2.2.3 · 8c8b4d36
  Tri Dao authored 1 year ago
  
  8c8b4d36
- Implement rotary embedding in flash_attn_with_kvcache · ccbb14f3
  Tri Dao authored 1 year ago
  
  ccbb14f3
- [CE] Implement CrossEntropyLoss in Triton · 5400fdc4
  Tri Dao authored 1 year ago
  
  5400fdc4
13 Sep, 2023 2 commits
- Simplify the implementation of KVcache attn by appending KV first · 56b7fc6e
  Tri Dao authored 1 year ago
  
  56b7fc6e
- Add tests for Pythia, GPT-JT, and RedPajama models · d0032700
  Tri Dao authored 1 year ago
  
  d0032700
12 Sep, 2023 1 commit
- Remove some unused headers · bb9beb36
  Tri Dao authored 1 year ago
  
  bb9beb36
11 Sep, 2023 4 commits
- Bump to v2.2.2 · 08c295c0
  Tri Dao authored 1 year ago
  
  08c295c0
- Swap seqlen_q and nheads for MQA to speed it up (h/t Daniel Haziza) · ee77b931
  Tri Dao authored 1 year ago
  
  ee77b931
- Add BigCode converters (#532) · 07005806
  Kevin Hu authored 1 year ago
  
  07005806
- [Gen] Fix calling update_graph_cache in tests · 8a733cbd
  Tri Dao authored 1 year ago
  
  8a733cbd
09 Sep, 2023 1 commit
- Inverse state dict for BERT (#527) · 4c91621a
  Kevin Hu authored 1 year ago
  
  4c91621a
07 Sep, 2023 1 commit
- [Gen] Use flash_attn_with_kvcache in generation · a86442f0
  Tri Dao authored 1 year ago
  
  a86442f0
06 Sep, 2023 2 commits
- Bump to v2.2.1 · a1576ad1
  Tri Dao authored 1 year ago
  
  a1576ad1
- [Rotary] Set device before launching Triton kernel to avoid error · 97951590
  Tri Dao authored 1 year ago
  
  97951590

GitLab

Menu