add get_tok_ids

Author: qmkakaxi

vllm_ascend/models/deepseek_v2.py

@@ -810,6 +810,9 @@ def get_all_expert_map(self,num_moe_layers):
 
         return torch.stack(all_loads, dim=0)
 
+    def get_topk_ids(self,layer_id):
+        return self.model.layers[layer_id+3].mlp.experts.topk_ids
+        
     def get_all_moe_loads(
         self,
         num_moe_layers: int,