From 0944107e02eeae8db9f68216c775ddaac99c4f95 Mon Sep 17 00:00:00 2001 From: carlushuang Date: Mon, 24 Feb 2025 00:18:16 +0800 Subject: [PATCH] hint mp --- example/ck_tile/13_moe_sorting/moe_sorting.cpp | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/example/ck_tile/13_moe_sorting/moe_sorting.cpp b/example/ck_tile/13_moe_sorting/moe_sorting.cpp index 872f0e6761..f00d948f25 100644 --- a/example/ck_tile/13_moe_sorting/moe_sorting.cpp +++ b/example/ck_tile/13_moe_sorting/moe_sorting.cpp @@ -237,12 +237,13 @@ bool test_moe_sorting(ck_tile::ArgParser args) } #endif - printf("[%s|%s]tokens:%d, num_experts:%d, topk:%d, ", + printf("[%s|%s]tokens:%d, num_experts:%d, topk:%d, mp:%d, ", index_prec.c_str(), weight_prec.c_str(), tokens, num_experts, - topk); + topk, + workspace_size != 0 ? 1 : 0); if(local_expert_masking) {