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