| .. |
|
blocksparse_attention
|
a2d476183f
fix: remove scipy and re-implement CSR matrix
|
7 месяцев назад |
|
__init__.py
|
9d81716bfd
[v0.5.3] Release Candidate (#388)
|
10 месяцев назад |
|
ipex_attn.py
|
805fa8721d
feat: use intel_extension_for_pytorch for CPU backend
|
7 месяцев назад |
|
paged_attn.py
|
156f577f79
feat: switch from `PYBIND11_MODULE` to `TORCH_LIBRARY` (#569)
|
7 месяцев назад |
|
prefix_prefill.py
|
d8f9f0ec16
fix: prefix prefill kernels for fp32 data type
|
7 месяцев назад |
|
triton_flash_attn.py
|
8e11259e90
missing triton autoconfig for rocm flash attn
|
8 месяцев назад |
AlpinDale
d8f9f0ec16
fix: prefix prefill kernels for fp32 data type