flash_attn -> vllm_flash_attn

6ac8e63a · Woosuk Kwon · 47ad0761 · 6ac8e63a · 6ac8e63a · 6ac8e63a
Commit 6ac8e63a authored Mar 28, 2024 by Woosuk Kwon
8 changed files
--- a/flash_attn/ops/triton/mlp.py
+++ b/flash_attn/ops/triton/mlp.py
--- a/flash_attn/ops/triton/rotary.py
+++ b/flash_attn/ops/triton/rotary.py
--- a/flash_attn/pyproject.toml
+++ b/flash_attn/pyproject.toml
--- a/flash_attn/utils/__init__.py
+++ b/flash_attn/utils/__init__.py
--- a/flash_attn/utils/benchmark.py
+++ b/flash_attn/utils/benchmark.py
--- a/flash_attn/utils/distributed.py
+++ b/flash_attn/utils/distributed.py
--- a/flash_attn/utils/generation.py
+++ b/flash_attn/utils/generation.py
--- a/flash_attn/utils/pretrained.py
+++ b/flash_attn/utils/pretrained.py