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

删除重复代码

parent ce1d4073
...@@ -2201,9 +2201,6 @@ class TritonExperts(mk.FusedMoEPermuteExpertsUnpermute): ...@@ -2201,9 +2201,6 @@ class TritonExperts(mk.FusedMoEPermuteExpertsUnpermute):
E, num_tokens, N, K, top_k_num = self.moe_problem_size( E, num_tokens, N, K, top_k_num = self.moe_problem_size(
hidden_states, w1, w2, topk_ids hidden_states, w1, w2, topk_ids
) )
if use_nn_moe:
N = w1.size(-1)
if use_nn_moe: if use_nn_moe:
N = w1.size(-1) 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