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

[310p] Add a PyTorch implementation of the GDN gating operator on 310P (#7430)

Browse Source 
### What this PR does / why we need it?
RFC #7394
Add a PyTorch implementation of the GDN gating operator 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>
This commit is contained in:
Shaoxu Cheng
2026-03-23 20:26:39 +08:00
committed by GitHub
parent e344a53127
commit 13397e9cb7
3 changed files with 116 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
vllm_ascend/_310p/ops/fla/__init__.py Normal file
View File

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