xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-04-11 21:07:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

hacks

Browse Source 
Signed-off-by: Bill Nell <bnell@redhat.com>
This commit is contained in:
Bill Nell 2025-05-30 02:33:58 +00:00
parent a0efd3106c
commit 95c40f9b09
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
vllm/model_executor/layers/fused_moe/fused_batched_moe.py
View File

@ -635,7 +635,8 @@ def batched_moe_kernel_quantize_input(
per_channel_quant: bool,
block_shape: Optional[list[int]] = None,
) -> tuple[torch.Tensor, Optional[torch.Tensor]]:
if (torch.compiler.is_compiling()
if (True or
torch.compiler.is_compiling()
or torch.cuda.is_current_stream_capturing()):
# Note: this does a bunch of extra work because expert_num_tokens is ignored
# but it does support torch.compile + cudagraphs.