Skip to content

Commit 6d7e988

Browse files
committed
fix
Signed-off-by: Bill Nell <bnell@redhat.com>
1 parent e1bc323 commit 6d7e988

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm/model_executor/layers/fused_moe/flashinfer_cutlass_moe.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -110,7 +110,7 @@ def workspace_shapes(
110110
"""
111111
workspace1 = (curr_M, K)
112112
workspace2 = (0, )
113-
output_shape = (M, K * 2 if if self.quant_dtype == "nvfp4" else K)
113+
output_shape = (M, K * 2 if self.quant_dtype == "nvfp4" else K)
114114
# The workspace is determined by `aq`, since it comes after any
115115
# potential communication op and is involved in the expert computation.
116116
return (workspace1, workspace2, output_shape)

0 commit comments

Comments
 (0)