TensorRT-LLMs/cpp/tensorrt_llm/pybind/batch_manager
pcastonguay 9643be5f20
[TRTLLM-5050][feat] Enable per-request stats with PyT backend (#4156)
* feat: Add per-request stats support with PyT backend

Signed-off-by: Patrice Castonguay <55748270+pcastonguay@users.noreply.github.com>

* Adding unit test

Signed-off-by: Patrice Castonguay <55748270+pcastonguay@users.noreply.github.com>

* Fixing stats unit test

Signed-off-by: Patrice Castonguay <55748270+pcastonguay@users.noreply.github.com>

* Fixing test with overlap

Signed-off-by: Patrice Castonguay <55748270+pcastonguay@users.noreply.github.com>

---------

Signed-off-by: Patrice Castonguay <55748270+pcastonguay@users.noreply.github.com>
2025-05-12 21:35:15 -04:00
..
algorithms.cpp [TRTLLM-3429] feat: Overlap scheduling in C++ runtime (#3625) 2025-05-06 15:06:46 +02:00
algorithms.h Update TensorRT-LLM (#2413) 2024-11-05 16:27:06 +08:00
bindings.cpp [TRTLLM-5050][feat] Enable per-request stats with PyT backend (#4156) 2025-05-12 21:35:15 -04:00
bindings.h Update TensorRT-LLM (#2413) 2024-11-05 16:27:06 +08:00
buffers.cpp Update TensorRT-LLM (#2849) 2025-03-04 18:44:00 +08:00
buffers.h Update TensorRT-LLM (#2755) 2025-02-11 03:01:00 +00:00
cacheTransceiver.cpp cacheTransceiver buffer manager (#3798) 2025-04-27 11:48:15 +08:00
cacheTransceiver.h Update TensorRT-LLM (#2820) 2025-02-25 21:21:49 +08:00
kvCacheManager.cpp cacheTransceiver buffer manager (#3798) 2025-04-27 11:48:15 +08:00
kvCacheManager.h Update TensorRT-LLM (#2413) 2024-11-05 16:27:06 +08:00
llmRequest.cpp feat: Add multimodal embedding field in LlmRequest (#3855) 2025-05-01 12:23:30 +08:00
llmRequest.h feat: Add multimodal embedding field in LlmRequest (#3855) 2025-05-01 12:23:30 +08:00