hf_text-generation-inference/backends/trtllm
Morgan Funtowicz b653605e54 feat(trtllm): fix logits retrieval 2024-12-10 23:28:13 +01:00
..
cmake feat(trtllm): expose finish reason to Rust 2024-12-10 16:51:22 +01:00
csrc feat(trtllm): fix logits retrieval 2024-12-10 23:28:13 +01:00
scripts chore(trtllm): update dependency towards 0.15.0 2024-12-04 12:02:42 +01:00
src feat(trtllm): expose finish reason to Rust 2024-12-10 16:51:22 +01:00
tests feat(trtllm): expose finish reason to Rust 2024-12-10 16:51:22 +01:00
CMakeLists.txt feat(trtllm): expose finish reason to Rust 2024-12-10 16:51:22 +01:00
Cargo.toml feat(backend): fix main.rs retrieving the tokenizer 2024-12-03 12:11:17 +01:00
README.md Rebase TRT-llm (#2331) 2024-07-31 10:33:10 +02:00
build.rs feat(trtllm): expose finish reason to Rust 2024-12-10 16:51:22 +01:00

README.md

Text Generation Inference - TensorRT-LLM Backend Implementation

Description

This folder provides the sources of the TensorRT-LLM backend implementation powered by TensorRT-LLM Executor new API

Simplified Request Sequence

sequenceDiagram
    actor User
    participant TextGenerationInference.HttpServer
    participant TextGenerationInference.TensorRtLlmBackend
    participant TextGenerationInference.TensorRtLlmWorkerThread
    participant TensorRtLlm.Executor
    participant Nvidia.Gpu
    User ->> TextGenerationInference.HttpServer: POST /generate
    TextGenerationInference.HttpServer ->> TextGenerationInference.TensorRtLlmBackend: Validate and forward inputs & parameters
    TextGenerationInference.TensorRtLlmBackend ->> TextGenerationInference.TensorRtLlmWorkerThread: Allocate a new context and spawn a new thread to handle the request
    TextGenerationInference.TensorRtLlmWorkerThread ->> TensorRtLlm.Executor: Submit the request to the In-Flight Batcher
    activate Nvidia.Gpu
    TensorRtLlm.Executor ->> Nvidia.Gpu: Add the request to the poll for execution
    TensorRtLlm.Executor -->> TextGenerationInference.TensorRtLlmWorkerThread: Response with an unique request identifier
    rect rgb(10, 92, 54)
        loop every 100us
            rect rgb(15, 81, 50)
                alt Acquire lock to query executor
                    TextGenerationInference.TensorRtLlmWorkerThread ->> TensorRtLlm.Executor: Poll request number of new token(s) generated
                else There are new generated tokens
                    TextGenerationInference.TensorRtLlmWorkerThread ->> TensorRtLlm.Executor: Retrieve newly generated tokens
                    TensorRtLlm.Executor -->> TextGenerationInference.TensorRtLlmWorkerThread: Return decoded token information and potential error (omitted)
                    rect rgb(11, 110, 79)
                        alt Generated token is final
                            TensorRtLlm.Executor ->> Nvidia.Gpu: Remove request from the scheduler and from the GPU
                            TextGenerationInference.TensorRtLlmWorkerThread -->> User: Stream the remaining decoded tokens and flush the connection
                        else Generated token is not final
                            TextGenerationInference.TensorRtLlmWorkerThread -->> User: Stream token back to the user as they get decoded
                        end
                    end
                end
            end
            deactivate Nvidia.Gpu
        end
    end