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 месяцев назад