[Bugfix] fix fused_moe test (#67353)

penPenf28 · web-flow · commit 2f9dc931325f · 2024-08-13T02:00:55.000-07:00
diff --git a/test/legacy_test/test_fused_moe_op.py b/test/legacy_test/test_fused_moe_op.py
@@ -131,7 +131,7 @@ def GetFusedMoeOut(self, tensor_x):
             None if self.quant_method == "None" else self.scale0,
             None if self.quant_method == "None" else self.scale1,
             self.quant_method,
-            2,
+            self.top_k,
         )
 
         return fused_out

Original file line number	Diff line number	Diff line change
`@@ -131,7 +131,7 @@ def GetFusedMoeOut(self, tensor_x):`
`131`	`131`	`None if self.quant_method == "None" else self.scale0,`
`132`	`132`	`None if self.quant_method == "None" else self.scale1,`
`133`	`133`	`self.quant_method,`
`134`		`- 2,`
	`134`	`+ self.top_k,`
`135`	`135`	`)`
`136`	`136`
`137`	`137`	`return fused_out`