Skip to content

Commit 4c998c3

Browse files
authored
[Code Simplification] delete cum_offsets_out (#3815)
* fix * fix
1 parent 0a1ce61 commit 4c998c3

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

custom_ops/gpu_ops/speculate_decoding/speculate_get_padding_offset.cu

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -139,6 +139,7 @@ std::vector<paddle::DataType> SpeculateGetPaddingOffsetInferDtype(
139139
PD_BUILD_STATIC_OP(speculate_get_padding_offset)
140140
.Inputs({"input_ids",
141141
"draft_tokens",
142+
"cum_offsets"
142143
"token_num",
143144
"seq_len",
144145
"seq_lens_encoder"})

0 commit comments

Comments
 (0)