Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "e288df0632d5bdde76c20bed8310b46d35b8e5ac"
Commit b59e0dd7 authored by zhuwenwen's avatar zhuwenwen
Browse files

fix cdiv params

parent 15d3c71a
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
vllm/model_executor/layers/fused_moe/fused_moe.py
View file @ b59e0dd7
...@@ -793,7 +793,7 @@ def invoke_fused_moe_triton_kernel( ...@@ -793,7 +793,7 @@ def invoke_fused_moe_triton_kernel(
EM = num_tokens * config["BLOCK_SIZE_M"] EM = num_tokens * config["BLOCK_SIZE_M"]
grid = lambda META: ( grid = lambda META: (
triton.cdiv(EM, META["BLOCK_SIZE_M"]) triton.cdiv(EM, META["BLOCK_SIZE_M"])
* triton.cdiv(B.size(1)), * triton.cdiv(B.size(1), META["BLOCK_SIZE_N"]),
) )
HAS_BIAS = B_bias is not None HAS_BIAS = B_bias is not None
......
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