Skip to content

Commit edf3b4c

Browse files
fix typo (#635)
1 parent 78ee163 commit edf3b4c

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

examples/deepseek_mla/example_mla_decode_persistent.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -106,8 +106,7 @@ def main_split_persistent(
106106
T.copy(acc_s, S_shared)
107107
T.copy(S_shared, acc_s_cast)
108108
for i in T.Parallel(block_H):
109-
logsum[i] = logsum[i] * sco
110-
es_scale[i] + scores_sum[i]
109+
logsum[i] = logsum[i] * scores_scale[i] + scores_sum[i]
111110
for i, j in T.Parallel(block_H, dim):
112111
acc_o[i, j] *= scores_scale[i]
113112
T.gemm(acc_s_cast, KV_shared, acc_o, policy=T.GemmWarpPolicy.FullCol)

0 commit comments

Comments
 (0)