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
Hide 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,18 +116,19 @@ bool test_moe_sorting(ck_tile::ArgParser args) ...@@ -116,18 +116,19 @@ 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{
weights_dev.GetDeviceBuffer(), topk_ids_dev.GetDeviceBuffer(),
sorted_ids_dev.GetDeviceBuffer(), weights_dev.GetDeviceBuffer(),
sorted_weights_dev.GetDeviceBuffer(), sorted_ids_dev.GetDeviceBuffer(),
expert_ids_dev.GetDeviceBuffer(), sorted_weights_dev.GetDeviceBuffer(),
sorted_id_cnt_dev.GetDeviceBuffer(), expert_ids_dev.GetDeviceBuffer(),
moe_buf_dev.GetDeviceBuffer(), sorted_id_cnt_dev.GetDeviceBuffer(),
tokens, moe_buf_dev.GetDeviceBuffer(),
unit_size, tokens,
experts, unit_size,
topk, experts,
static_cast<ck_tile::index_t>(moe_buf_host.get_element_space_size_in_bytes())}; topk,
static_cast<ck_tile::index_t>(moe_buf_host.get_element_space_size_in_bytes())};
ck_tile::stream_config sc{nullptr, ck_tile::stream_config sc{nullptr,
true, true,
......
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