[Tiny fix] Super tiny fix mul_add naive forward bug (#18964)

2026-02-18 16:18:43 +08:00
parent 513c12d23f
commit 390c154306
1 changed files with 1 additions and 1 deletions
--- a/python/sglang/multimodal_gen/runtime/layers/elementwise.py
+++ b/python/sglang/multimodal_gen/runtime/layers/elementwise.py
@@ -23,7 +23,7 @@ class MulAdd(CustomOp):
            num_frames = b.shape[1]
            frame_seqlen = a.shape[1] // num_frames
            return c + (
-                a.unflatten(dim=1, sizes=(num_frames, frame_seqlen)) * b
+                a.unflatten(dim=1, sizes=(num_frames, frame_seqlen)) * (k + b)
            ).flatten(1, 2)
        else:
            # b.shape: [batch_size, 1, inner_dim]