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