flash_attn -> vllm_flash_attn

6ac8e63a · Woosuk Kwon · 47ad0761 · 6ac8e63a · 6ac8e63a · 6ac8e63a
Commit 6ac8e63a authored Mar 28, 2024 by Woosuk Kwon
20 changed files
--- a/flash_attn/__init__.py
+++ b/flash_attn/__init__.py
--- a/flash_attn/bert_padding.py
+++ b/flash_attn/bert_padding.py
--- a/flash_attn/flash_attn_interface.py
+++ b/flash_attn/flash_attn_interface.py
--- a/flash_attn/flash_attn_triton.py
+++ b/flash_attn/flash_attn_triton.py
--- a/flash_attn/flash_attn_triton_og.py
+++ b/flash_attn/flash_attn_triton_og.py
--- a/flash_attn/flash_blocksparse_attention.py
+++ b/flash_attn/flash_blocksparse_attention.py
--- a/flash_attn/flash_blocksparse_attn_interface.py
+++ b/flash_attn/flash_blocksparse_attn_interface.py
--- a/flash_attn/fused_softmax.py
+++ b/flash_attn/fused_softmax.py
--- a/flash_attn/layers/__init__.py
+++ b/flash_attn/layers/__init__.py
--- a/flash_attn/layers/patch_embed.py
+++ b/flash_attn/layers/patch_embed.py
--- a/flash_attn/layers/rotary.py
+++ b/flash_attn/layers/rotary.py
--- a/flash_attn/losses/__init__.py
+++ b/flash_attn/losses/__init__.py
--- a/flash_attn/losses/cross_entropy.py
+++ b/flash_attn/losses/cross_entropy.py
--- a/flash_attn/models/__init__.py
+++ b/flash_attn/models/__init__.py
--- a/flash_attn/models/baichuan.py
+++ b/flash_attn/models/baichuan.py
--- a/flash_attn/models/bert.py
+++ b/flash_attn/models/bert.py
--- a/flash_attn/models/bigcode.py
+++ b/flash_attn/models/bigcode.py
--- a/flash_attn/models/btlm.py
+++ b/flash_attn/models/btlm.py
--- a/flash_attn/models/falcon.py
+++ b/flash_attn/models/falcon.py
--- a/flash_attn/models/gpt.py
+++ b/flash_attn/models/gpt.py