| .. |
|
__init__.py
|
9d81716bfd
[v0.5.3] Release Candidate (#388)
|
8 ヶ月 前 |
|
abstract.py
|
22305c91e9
refactor _prepare_model_input_tensor and attn metadata builder for most backends
|
5 ヶ月 前 |
|
blocksparse_attn.py
|
22305c91e9
refactor _prepare_model_input_tensor and attn metadata builder for most backends
|
5 ヶ月 前 |
|
flash_attn.py
|
22305c91e9
refactor _prepare_model_input_tensor and attn metadata builder for most backends
|
5 ヶ月 前 |
|
flashinfer.py
|
8adc496a2a
fix: use paged attention for bloc swapping/copying in flashinfer
|
5 ヶ月 前 |
|
ipex_attn.py
|
22305c91e9
refactor _prepare_model_input_tensor and attn metadata builder for most backends
|
5 ヶ月 前 |
|
openvino.py
|
0886c361f4
feat: OpenVINO CPU backend (#576)
|
5 ヶ月 前 |
|
pallas.py
|
9d7beaa5b9
chore: separate kv_scale into k_scale and v_scale
|
5 ヶ月 前 |
|
rocm_flash_attn.py
|
fa15bad2ea
chore: minor AMD fixes
|
5 ヶ月 前 |
|
torch_sdpa.py
|
22305c91e9
refactor _prepare_model_input_tensor and attn metadata builder for most backends
|
5 ヶ月 前 |
|
utils.py
|
22305c91e9
refactor _prepare_model_input_tensor and attn metadata builder for most backends
|
5 ヶ月 前 |
|
xformers.py
|
22305c91e9
refactor _prepare_model_input_tensor and attn metadata builder for most backends
|
5 ヶ月 前 |
AlpinDale
fa15bad2ea
chore: minor AMD fixes