Skip to content

GitLab

Commit 7f775ad2 authored by 王敏's avatar 王敏
Browse files

修复错误代码

parent d0dafaf4
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
csrc/moe/torch_bindings.cpp
View file @ 7f775ad2
...@@ -22,13 +22,13 @@ TORCH_LIBRARY_EXPAND(TORCH_EXTENSION_NAME, m) { ...@@ -22,13 +22,13 @@ TORCH_LIBRARY_EXPAND(TORCH_EXTENSION_NAME, m) {
" Tensor! num_tokens_post_pad) -> ()"); " Tensor! num_tokens_post_pad) -> ()");
m.impl("moe_align_block_size", torch::kCUDA, &moe_align_block_size); m.impl("moe_align_block_size", torch::kCUDA, &moe_align_block_size);
ops.def( m.def(
"ep_moe_align_block_size(Tensor topk_ids, int num_experts," "ep_moe_align_block_size(Tensor topk_ids, int num_experts,"
" int block_size, Tensor! sorted_token_ids," " int block_size, Tensor! sorted_token_ids,"
" Tensor! experts_ids," " Tensor! experts_ids,"
" Tensor! num_tokens_post_pad," " Tensor! num_tokens_post_pad,"
" int start_expert, int end_expert) -> ()"); " int start_expert, int end_expert) -> ()");
ops.impl("ep_moe_align_block_size", torch::kCUDA, &ep_moe_align_block_size); m.impl("ep_moe_align_block_size", torch::kCUDA, &ep_moe_align_block_size);
// temporarily adapted from // temporarily adapted from
// https://github.com/sgl-project/sglang/commit/ded9fcd09a43d5e7d5bb31a2bc3e9fc21bf65d2a // https://github.com/sgl-project/sglang/commit/ded9fcd09a43d5e7d5bb31a2bc3e9fc21bf65d2a
......
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