[Bugfix] Fix _fused_moe_lora_expand signature mismatch (#33821)

Signed-off-by: Xin Yang <xyangx@amazon.com>
This commit is contained in:
Xin Yang
2026-02-06 18:45:59 -08:00
committed by GitHub
parent ec28784fdc
commit 3920cafdd6

View File

@@ -779,7 +779,6 @@ def _fused_moe_lora_shrink_fake(
def _fused_moe_lora_expand_fake(
output: torch.Tensor,
a_intermediate_cache1: torch.Tensor,
b_intermediate_cache1: torch.Tensor,
lora_b_stacked: list[torch.Tensor],
topk_weights: torch.Tensor,
sorted_token_ids: torch.Tensor | None,