Commit aec90b84 authored by 王敏's avatar 王敏
Browse files

删除重复代码

parent ce1d4073
......@@ -2201,9 +2201,6 @@ class TritonExperts(mk.FusedMoEPermuteExpertsUnpermute):
E, num_tokens, N, K, top_k_num = self.moe_problem_size(
hidden_states, w1, w2, topk_ids
)
if use_nn_moe:
N = w1.size(-1)
if use_nn_moe:
N = w1.size(-1)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment