Skip to content

Commit a000ab6

Browse files
authored
fix sp_pad_copy.
1 parent c7cb0d6 commit a000ab6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

lightllm/common/basemodel/triton_kernel/sp_pad_copy.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
# if token_num % sp_world_size != 0, will pad
44
# token to token_num % sp_world_size == 0, this.
55
# demo: token_num = 3, sp_world_size = 4, will get
6-
# ans_tensor shape is [4, hidden_dim].
6+
# ans_tensor shape is [1, hidden_dim].
77
import torch
88

99
import triton

0 commit comments

Comments
 (0)