[opt kimi k2 1 / n] Add kimi k2 moe fused gate (#13287)
This commit is contained in:
@@ -319,6 +319,7 @@ set(SOURCES
|
||||
"csrc/moe/marlin_moe_wna16/ops.cu"
|
||||
"csrc/moe/moe_align_kernel.cu"
|
||||
"csrc/moe/moe_fused_gate.cu"
|
||||
"csrc/moe/kimi_k2_moe_fused_gate.cu"
|
||||
"csrc/moe/moe_sum.cu"
|
||||
"csrc/moe/moe_sum_reduce.cu"
|
||||
"csrc/moe/moe_topk_softmax_kernels.cu"
|
||||
|
||||
Reference in New Issue
Block a user