EngineX/xc-llm-ascend
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[310P]fused recurrent gated delta rule pytorch core and ut (#7398)

Browse Source 
### What this PR does / why we need it?
RFC https://github.com/vllm-project/vllm-ascend/issues/7394
Add a PyTorch implementation of the fused recurrent gated delta ruler on
310P.
### Does this PR introduce _any_ user-facing change?
NO
### How was this patch tested?
UT
- vLLM version: v0.17.0
- vLLM main:
4497431df6

---------

Signed-off-by: Tflowers-0129 <2906339855@qq.com>
Co-authored-by: wangxiyuan <wangxiyuan1007@gmail.com>
This commit is contained in:
Shaoxu Cheng
2026-03-25 08:53:14 +08:00
committed by GitHub
parent 54879467c4
commit 3f4087a8f0
3 changed files with 309 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
vllm_ascend/_310p/ops/fla/__init__.py
View File

@@ -1,3 +1,7 @@
from .fused_gdn_gating import fused_gdn_gating_pytorch
from .fused_recurrent_gated_delta_rule import fused_recurrent_gated_delta_rule_pytorch
__all__ = ["fused_gdn_gating_pytorch"]
__all__ = [
"fused_gdn_gating_pytorch",
"fused_recurrent_gated_delta_rule_pytorch",
]