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 달 전