| .. |
|
blocksparse_attention
|
a2d476183f
fix: remove scipy and re-implement CSR matrix
|
il y a 5 mois |
|
__init__.py
|
9d81716bfd
[v0.5.3] Release Candidate (#388)
|
il y a 8 mois |
|
ipex_attn.py
|
9d7beaa5b9
chore: separate kv_scale into k_scale and v_scale
|
il y a 5 mois |
|
paged_attn.py
|
9d7beaa5b9
chore: separate kv_scale into k_scale and v_scale
|
il y a 5 mois |
|
prefix_prefill.py
|
d8f9f0ec16
fix: prefix prefill kernels for fp32 data type
|
il y a 5 mois |
|
triton_flash_attn.py
|
8e11259e90
missing triton autoconfig for rocm flash attn
|
il y a 6 mois |
AlpinDale
9d7beaa5b9
chore: separate kv_scale into k_scale and v_scale