"vscode:/vscode.git/clone" did not exist on "acf9f945ed272fbb650ef71cf582bd7e48b52d06"
Unverified Commit fd455508 authored by Andrey Khalyavin's avatar Andrey Khalyavin Committed by GitHub
Browse files

[BugFix] Fix misprint introduced by modular_kernel refactoring. (#28728)


Signed-off-by: default avatarAndrey Khalyavin <halyavin@yandex-team.ru>
parent cec275ef
......@@ -1060,7 +1060,7 @@ class FusedMoEModularKernel(torch.nn.Module):
global_num_experts=global_num_experts,
expert_map=expert_map,
a1q_scale=_slice_scales(a1q_scale, s, e),
a2_scale=_slice_scales(self.fused_experts.a2_scale, e, e),
a2_scale=_slice_scales(self.fused_experts.a2_scale, s, e),
workspace13=workspace13,
workspace2=workspace2,
expert_tokens_meta=c_expert_tokens_meta,
......
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