.. |
__init__.py
|
9d81716bfd
[v0.5.3] Release Candidate (#388)
|
10 bulan lalu |
abstract.py
|
696f2cd59c
add phi3_small support with blocksparse attention
|
7 bulan lalu |
blocksparse_attn.py
|
696f2cd59c
add phi3_small support with blocksparse attention
|
7 bulan lalu |
flash_attn.py
|
156f577f79
feat: switch from `PYBIND11_MODULE` to `TORCH_LIBRARY` (#569)
|
7 bulan lalu |
flashinfer.py
|
156f577f79
feat: switch from `PYBIND11_MODULE` to `TORCH_LIBRARY` (#569)
|
7 bulan lalu |
rocm_flash_attn.py
|
71a26f0998
chore: use pytorch sdpa backend to do naive attention for rocm
|
7 bulan lalu |
torch_sdpa.py
|
696f2cd59c
add phi3_small support with blocksparse attention
|
7 bulan lalu |
xformers.py
|
696f2cd59c
add phi3_small support with blocksparse attention
|
7 bulan lalu |