Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
change
sglang
Commits
8e7b3154
"vscode:/vscode.git/clone" did not exist on "029cb3fe4526084172c30be14278d46ecd5bf17c"
Unverified
Commit
8e7b3154
authored
Mar 30, 2025
by
yinfan98
Committed by
GitHub
Mar 29, 2025
Browse files
quick fix: add default for new kernel (#4898)
parent
45dcfc2e
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
sgl-kernel/python/sgl_kernel/moe.py
sgl-kernel/python/sgl_kernel/moe.py
+1
-1
No files found.
sgl-kernel/python/sgl_kernel/moe.py
View file @
8e7b3154
...
...
@@ -41,6 +41,6 @@ def moe_fused_gate(input_tensor, bias, num_expert_group, topk_group, topk):
# the #experts is decided by the input tensor shape and we currently only support power of 2 #experts
# and #experts should be divisible by num_expert_group. #expert/num_expert_group <= 32 is limitted for now.
# for non-supported case, we suggestion to use the biased_grouped_topk func in sglang.srt.layers.moe.topk
return
torch
.
ops
.
sgl_kernel
.
moe_fused_gate
(
return
torch
.
ops
.
sgl_kernel
.
moe_fused_gate
.
default
(
input_tensor
,
bias
,
num_expert_group
,
topk_group
,
topk
)
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment