Skip to content

Commit 0356c38

Browse files
XuZhang99Kang-Meng
authored andcommitted
bugfix: ensure running_sequences size is greater than 0.
1 parent b77b20e commit 0356c38

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

xllm/core/framework/batch/batch_factory.cpp

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -69,10 +69,14 @@ std::vector<Batch> BatchFactory::create_batches(
6969
COUNTER_ADD(num_processing_tokens_total_prompt, num_prompt_tokens);
7070
COUNTER_ADD(num_processing_tokens_total_generated, num_generated_tokens);
7171

72-
HISTOGRAM_OBSERVE(num_prompt_tokens_per_request,
73-
num_prompt_tokens / running_sequences.size());
74-
HISTOGRAM_OBSERVE(num_generated_tokens_per_request,
75-
num_generated_tokens / running_sequences.size());
72+
if (running_sequences.size() > 0) {
73+
HISTOGRAM_OBSERVE(
74+
num_prompt_tokens_per_request,
75+
static_cast<int64_t>(num_prompt_tokens / running_sequences.size()));
76+
HISTOGRAM_OBSERVE(
77+
num_generated_tokens_per_request,
78+
static_cast<int64_t>(num_generated_tokens / running_sequences.size()));
79+
}
7680

7781
return batches;
7882
}

0 commit comments

Comments
 (0)