[Perf][Qwen3.5] Add case 512 to topkGatingSoftmaxKernelLauncher, (#25775)

2026-05-25 16:08:21 +08:00
parent b0cf01eb85
commit e27d4fb70f
3 changed files with 6 additions and 3 deletions
@@ -144,7 +144,7 @@ if IS_CI:
 else:
    num_tokens_range = [128, 512, 1024, 2048, 4096, 8192, 16384, 32768]
    num_experts_range = [32, 64, 128, 256, 12, 512]
-    topk_range = [1, 2, 4, 8]
+    topk_range = [1, 2, 4, 8, 10]

 configs = list(itertools.product(num_tokens_range, num_experts_range, topk_range))