Ryan bcdfc1d6b9 Add custom op declaration for all_reduce (#3473) ...

* add custom op declaration

* roll back try except

2025-08-20 20:29:58 +08:00

..

ops

make append_attn supports mask_offset (#3138)

2025-08-14 03:40:55 -07:00

__init__.py

[SOT] Mark dynamic dims by type annotations (#2771)

2025-07-22 00:23:52 -07:00

append_attn_backend.py

make append_attn supports mask_offset (#3138)

2025-08-14 03:40:55 -07:00

attention_selecter.py

polish code with new pre-commit rule (#2923)

2025-07-19 23:19:27 +08:00

attention.py

support qk norm (#3145)

2025-08-05 16:46:14 +08:00

base_attention_backend.py

[MetaxGPU] Support FastDeploy on metax gpu (#3241)

2025-08-13 11:11:54 +08:00

block_multihead_attn_backend.py

[Executor] Refactor GetBlockShapeAndSplitKVBlock Kernel (#2989)

2025-07-31 00:09:31 +08:00

flash_attn_backend.py

【Fix Bug】 修复 fa3 支持集中式bug (#3235)

2025-08-06 16:24:27 +08:00

iluvatar_attn_backend.py

[Iluvatar GPU] Optimze attention and moe performance (#3234)

2025-08-08 10:51:24 +08:00

mla_attention_backend.py

Add custom op declaration for all_reduce (#3473)

2025-08-20 20:29:58 +08:00

native_paddle_backend.py

polish code with new pre-commit rule (#2923)

2025-07-19 23:19:27 +08:00

utils.py

polish code with new pre-commit rule (#2923)

2025-07-19 23:19:27 +08:00

xpu_attn_backend.py

[Executor] Refactor GetBlockShapeAndSplitKVBlock Kernel (#2989)

2025-07-31 00:09:31 +08:00