Skip to content

GitLab

Commit 8a4a6fd8 authored by zhuwenwen's avatar zhuwenwen
Browse files

update fused_moe.py

parent f48aca62
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
vllm/model_executor/layers/fused_moe/fused_moe.py
View file @ 8a4a6fd8
...@@ -670,7 +670,7 @@ def fused_moe_kernel( ...@@ -670,7 +670,7 @@ def fused_moe_kernel(
BLOCK_SIZE_N: tl.constexpr, BLOCK_SIZE_N: tl.constexpr,
BLOCK_SIZE_K: tl.constexpr, BLOCK_SIZE_K: tl.constexpr,
GROUP_SIZE_M: tl.constexpr, GROUP_SIZE_M: tl.constexpr,
# SPLIT_K: tl.constexpr, SPLIT_K: tl.constexpr,
MUL_ROUTED_WEIGHT: tl.constexpr, MUL_ROUTED_WEIGHT: tl.constexpr,
top_k: tl.constexpr, top_k: tl.constexpr,
compute_type: tl.constexpr, compute_type: tl.constexpr,
......
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