[Ck_tile]fix moe_sorting fuse set_zero remod

0a9080a2 · dummycoderfe · 0385cded · 0a9080a2
Commit 0a9080a2 authored Nov 06, 2024 by dummycoderfe
Show whitespace changes
Inline Side-by-side

Showing with 13 additions and 12 deletions

example/ck_tile/13_moe_sorting/moe_sorting.cpp example/ck_tile/13_moe_sorting/moe_sorting.cpp +13 -12

No files found.
--- a/example/ck_tile/13_moe_sorting/moe_sorting.cpp
+++ b/example/ck_tile/13_moe_sorting/moe_sorting.cpp
@@ -116,7 +116,8 @@ bool test_moe_sorting(ck_tile::ArgParser args)

    moe_sorting_trait trait{index_prec, weight_prec, experts, topk, unit_size, tokens};

-    moe_sorting_args karg{topk_ids_dev.GetDeviceBuffer(),
+    moe_sorting_args karg{
+        topk_ids_dev.GetDeviceBuffer(),
        weights_dev.GetDeviceBuffer(),
        sorted_ids_dev.GetDeviceBuffer(),
        sorted_weights_dev.GetDeviceBuffer(),