Hongxia Yang 18d23f642a

[ROCm][Hardware][AMD] Enable group query attention for triton FA (#4406)

2024-04-26 23:37:40 -07:00

..

__init__.py

[Core] Refactor Attention Take 2 (#3462)

2024-03-25 04:39:33 +00:00

abstract.py

[Mypy] Part 3 fix typing for nested directories for most of directory (#4161)

2024-04-22 21:32:44 -07:00

flash_attn.py

[Core][5/N] Fully working chunked prefill e2e (#3884)

2024-04-10 17:56:48 -07:00

rocm_flash_attn.py

[ROCm][Hardware][AMD] Enable group query attention for triton FA (#4406)

2024-04-26 23:37:40 -07:00

torch_sdpa.py

[Mypy] Part 3 fix typing for nested directories for most of directory (#4161)

2024-04-22 21:32:44 -07:00

xformers.py

[Mypy] Part 3 fix typing for nested directories for most of directory (#4161)

2024-04-22 21:32:44 -07:00