fix bug for hard_gate_moe proto parse (bytedance#426)

Anychnn · web-flow · commit 1f6d693d0bc8 · 2022-11-24T21:54:21.000+08:00
diff --git a/lightseq/inference/proto/moe_weight.cc b/lightseq/inference/proto/moe_weight.cc
@@ -106,10 +106,10 @@ void MoeWeight<OpType_>::proto_get_model_config(const Moe &moe,
     std::vector<int> langs2gate_langs;
     std::vector<int> langs2gate_gates;
     for (int lang : moe.lang2gate().langs()) {
-      langs2gate_langs.push_back(id);
+      langs2gate_langs.push_back(lang);
     }
     for (int gate : moe.lang2gate().gates()) {
-      langs2gate_gates.push_back(id);
+      langs2gate_gates.push_back(gate);
     }
 
     for (int i = 0; i < langs2gate_langs.size(); i++) {

Original file line number	Diff line number	Diff line change
`@@ -106,10 +106,10 @@ void MoeWeight<OpType_>::proto_get_model_config(const Moe &moe,`
`106`	`106`	`std::vector<int> langs2gate_langs;`
`107`	`107`	`std::vector<int> langs2gate_gates;`
`108`	`108`	`for (int lang : moe.lang2gate().langs()) {`
`109`		`- langs2gate_langs.push_back(id);`
	`109`	`+ langs2gate_langs.push_back(lang);`
`110`	`110`	`}`
`111`	`111`	`for (int gate : moe.lang2gate().gates()) {`
`112`		`- langs2gate_gates.push_back(id);`
	`112`	`+ langs2gate_gates.push_back(gate);`
`113`	`113`	`}`
`114`	`114`
`115`	`115`	`for (int i = 0; i < langs2gate_langs.size(); i++) {`