Skip to content

Commit

Permalink
Rerun unit test
Browse files Browse the repository at this point in the history
Signed-off-by: Shaoting Feng <[email protected]>
  • Loading branch information
Shaoting-Feng committed Jan 24, 2025
1 parent 7549568 commit b8fb040
Showing 1 changed file with 1 addition and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -390,8 +390,7 @@ def build_partial_prefill_input(
rebuilt_slot_mapping.append(new_slot_mapping)
rebuilt_max_query_len = max(q_len, rebuilt_max_query_len)
last_query_start_loc += q_len
rebuilt_query_start_loc.append(
last_query_start_loc) # start with 0
rebuilt_query_start_loc.append(last_query_start_loc)
rebuilt_context_lens_tensor.append(num_computed_token)

# recover `block_table`
Expand Down

0 comments on commit b8fb040

Please sign in to comment.