From 4c998c363679a49ed57db3e18cfceb9f16f65ffc Mon Sep 17 00:00:00 2001 From: lizexu123 <39205361+lizexu123@users.noreply.github.com> Date: Wed, 3 Sep 2025 16:15:33 +0800 Subject: [PATCH] [Code Simplification] delete cum_offsets_out (#3815) * fix * fix --- .../gpu_ops/speculate_decoding/speculate_get_padding_offset.cu | 1 + 1 file changed, 1 insertion(+) diff --git a/custom_ops/gpu_ops/speculate_decoding/speculate_get_padding_offset.cu b/custom_ops/gpu_ops/speculate_decoding/speculate_get_padding_offset.cu index e37dacbf3..07f33ee2d 100644 --- a/custom_ops/gpu_ops/speculate_decoding/speculate_get_padding_offset.cu +++ b/custom_ops/gpu_ops/speculate_decoding/speculate_get_padding_offset.cu @@ -139,6 +139,7 @@ std::vector SpeculateGetPaddingOffsetInferDtype( PD_BUILD_STATIC_OP(speculate_get_padding_offset) .Inputs({"input_ids", "draft_tokens", + "cum_offsets" "token_num", "seq_len", "seq_lens_encoder"})