Fix overflow. Fix coverity. (openvinotoolkit#2179)

olpipi · Wovchena · web-flow · commit a9a4e4172ae1 · 2025-05-12T09:36:21.000Z
Fix overflow. Fix coverity.

CVS-165982

Co-authored-by: Vladimir Zlobin &lt;vladimir.zlobin@intel.com&gt;
diff --git a/src/cpp/src/continuous_batching/scheduler.hpp b/src/cpp/src/continuous_batching/scheduler.hpp
@@ -191,7 +191,7 @@ class Scheduler {
         if (tokens_in_last_block == 0) {
             tokens_in_last_block = block_size;
         }
-        preempted_tokens = tokens_in_last_block + std::max<size_t>((int)logical_blocks_released - 1, 0) * block_size;
+        preempted_tokens = tokens_in_last_block + (logical_blocks_released == 0 ? 0 : logical_blocks_released - 1) * block_size;
 
         // case when preemption requires preempt prompt tokens
         if (!m_config.dynamic_split_fuse && processed_tokens - preempted_tokens < sequence_group->get_prompt_len()) {

Original file line number	Diff line number	Diff line change
`@@ -191,7 +191,7 @@ class Scheduler {`
`191`	`191`	`if (tokens_in_last_block == 0) {`
`192`	`192`	`tokens_in_last_block = block_size;`
`193`	`193`	`}`
`194`		`- preempted_tokens = tokens_in_last_block + std::max<size_t>((int)logical_blocks_released - 1, 0) * block_size;`
	`194`	`+ preempted_tokens = tokens_in_last_block + (logical_blocks_released == 0 ? 0 : logical_blocks_released - 1) * block_size;`
`195`	`195`
`196`	`196`	`// case when preemption requires preempt prompt tokens`
`197`	`197`	`if (!m_config.dynamic_split_fuse && processed_tokens - preempted_tokens < sequence_group->get_prompt_len()) {`