Skip to content

GitLab

Commit 0a9080a2 authored by dummycoderfe's avatar dummycoderfe
Browse files

[Ck_tile]fix moe_sorting fuse set_zero remod

parent 0385cded
Show whitespace changes
Inline Side-by-side
Showing with 13 additions and 12 deletions
example/ck_tile/13_moe_sorting/moe_sorting.cpp
View file @ 0a9080a2
...@@ -116,7 +116,8 @@ bool test_moe_sorting(ck_tile::ArgParser args) ...@@ -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_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(), weights_dev.GetDeviceBuffer(),
sorted_ids_dev.GetDeviceBuffer(), sorted_ids_dev.GetDeviceBuffer(),
sorted_weights_dev.GetDeviceBuffer(), sorted_weights_dev.GetDeviceBuffer(),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment