Skip to content

Commit d0bf52a

Browse files
committed
add mhc npu
1 parent 3f85b72 commit d0bf52a

2 files changed

Lines changed: 1759 additions & 0 deletions

File tree

src/liger_kernel/ops/backends/_ascend/ops/__init__.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,9 @@
5252
from liger_kernel.ops.backends._ascend.ops.llama4_rope import LigerLlama4RopeFunction
5353
from liger_kernel.ops.backends._ascend.ops.llama4_rope import llama4_rope_backward
5454
from liger_kernel.ops.backends._ascend.ops.llama4_rope import llama4_rope_forward
55+
from liger_kernel.ops.backends._ascend.ops.mhc import LigerMHCCoeffsFunction
56+
from liger_kernel.ops.backends._ascend.ops.mhc import LigerMHCPostResFunction
57+
from liger_kernel.ops.backends._ascend.ops.mhc import LigerMHCPreFunction
5558
from liger_kernel.ops.backends._ascend.ops.poly_norm import LigerPolyNormFunction
5659
from liger_kernel.ops.backends._ascend.ops.poly_norm import poly_norm_backward
5760
from liger_kernel.ops.backends._ascend.ops.poly_norm import poly_norm_forward
@@ -140,4 +143,7 @@
140143
"sparsemax_backward",
141144
"LigerFusedNeighborhoodAttentionFunction",
142145
"fused_neighborhood_attention_forward",
146+
"LigerMHCCoeffsFunction",
147+
"LigerMHCPreFunction",
148+
"LigerMHCPostResFunction",
143149
]

0 commit comments

Comments
 (0)