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