add missing arg

Signed-off-by: vllmellm <vllm.ellm@embeddedllm.com>
This commit is contained in:
vllmellm 2025-11-03 07:09:52 +00:00
parent b13c4bb25c
commit 7794009661

View File

@ -22,6 +22,7 @@ def flashinfer_w8a8_scaled_mm(
As: torch.Tensor,
Bs: torch.Tensor,
bias: torch.Tensor,
output_shape: list,
) -> torch.Tensor:
return flashinfer_scaled_fp8_mm(
A, B, out_dtype=out_dtype, scale_a=As, scale_b=Bs, bias=bias