Skip to content

Commit 0838be5

Browse files
committed
fix: Correct enum definition for LlmRequestState
- Moved the kGENERATION_TO_COMPLETE state definition to maintain consistent ordering within the enum. - Ensured clarity in the state transitions for better code readability. Signed-off-by: Robin Kobus <[email protected]>
1 parent 133ec32 commit 0838be5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

cpp/include/tensorrt_llm/batch_manager/llmRequest.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,9 +58,9 @@ enum class LlmRequestState : int32_t
5858
/// used in layer-wise transmission
5959
kDISAGG_GENERATION_TRANS_COMPLETE = 12, ///< Kv cache transmission are finished
6060
kGENERATION_IN_PROGRESS = 13, ///< Generation phase is in progress
61-
kGENERATION_TO_COMPLETE = 14, ///< Generation phase is to be completed
6261

6362
// schedulable states ends
63+
kGENERATION_TO_COMPLETE = 14, ///< Generation phase is to be completed
6464
kGENERATION_COMPLETE = 20, ///< Generation phase completed
6565
kDISAGG_CONTEXT_TRANS_IN_PROGRESS = 21, ///< Waiting context-only request transmitting the kv cache,
6666
/// after computation finished

0 commit comments

Comments
 (0)