Update xllm/core/framework/block/block_manager_pool.cpp

RobbieLeung · gemini-code-assist[bot] · web-flow · commit 648094e56474 · 2026-03-31T18:37:33.000+08:00
Co-authored-by: gemini-code-assist[bot] &lt;176961590+gemini-code-assist[bot]@users.noreply.github.com&gt;
diff --git a/xllm/core/framework/block/block_manager_pool.cpp b/xllm/core/framework/block/block_manager_pool.cpp
@@ -218,10 +218,10 @@ bool BlockManagerPool::allocate(Sequence* sequence, size_t num_tokens) {
   if (blocks.size() != num_additional_blocks) {
     if (started_empty) {
       block_managers_[dp_rank]->deallocate(sequence->kv_state().kv_blocks());
-      sequence->reset();
       if (needs_embedding_id) {
         deallocate_embedding_id(sequence, dp_rank);
       }
+      sequence->reset();
     }
     // LOG(ERROR) << " Fail to allocate " << num_additional_blocks << "
     // blocks.";

Original file line number	Diff line number	Diff line change
`@@ -218,10 +218,10 @@ bool BlockManagerPool::allocate(Sequence* sequence, size_t num_tokens) {`
`218`	`218`	`if (blocks.size() != num_additional_blocks) {`
`219`	`219`	`if (started_empty) {`
`220`	`220`	`block_managers_[dp_rank]->deallocate(sequence->kv_state().kv_blocks());`
`221`		`- sequence->reset();`
`222`	`221`	`if (needs_embedding_id) {`
`223`	`222`	`deallocate_embedding_id(sequence, dp_rank);`
`224`	`223`	`}`
	`224`	`+ sequence->reset();`
`225`	`225`	`}`
`226`	`226`	`// LOG(ERROR) << " Fail to allocate " << num_additional_blocks << "`
`227`	`227`	`// blocks.";`