Unverified Commit 3920cafd authored by Xin Yang's avatar Xin Yang Committed by GitHub
Browse files

[Bugfix] Fix _fused_moe_lora_expand signature mismatch (#33821)


Signed-off-by: default avatarXin Yang <xyangx@amazon.com>
parent ec28784f
......@@ -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,
......
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