TensorRT-LLMs/cpp/tests/batch_manager
Kate Cheng 7dbe618683
feat: Add multimodal embedding field in LlmRequest (#3855)
* Add a new param to LlmRequest and Request to natively support mm

Signed-off-by: Kate Cheng <yunhsuanc@nvidia.com>

* update comment

Signed-off-by: Kate Cheng <yunhsuanc@nvidia.com>

* Update tests to match the new LlmRequest constructor parameters

Signed-off-by: Kate Cheng <yunhsuanc@nvidia.com>

* Modify unitTest and modify mm_embeding's dict name in llama4

Signed-off-by: Kate Cheng <yunhsuanc@nvidia.com>

* Fix based on comments

Signed-off-by: Kate Cheng <yunhsuanc@nvidia.com>

* Fix comment

Signed-off-by: Kate Cheng <yunhsuanc@nvidia.com>

* Fix LlmRequest initialization in kvCacheManagerTest

Signed-off-by: Kate Cheng <yunhsuanc@nvidia.com>

* Clean up code for promt_tuning_config

Signed-off-by: Kate Cheng <yunhsuanc@nvidia.com>

* Clean up prompt_tuning_config in GenerationRequest

Signed-off-by: Kate Cheng <yunhsuanc@nvidia.com>

---------

Signed-off-by: Kate Cheng <yunhsuanc@nvidia.com>
Co-authored-by: Haohang Huang <31998628+symphonylyh@users.noreply.github.com>
2025-05-01 12:23:30 +08:00
..
blockKeyTest.cpp fix partialMatch (#3413) 2025-04-11 16:42:52 +08:00
cacheTransceiverTest.cpp cacheTransceiver buffer manager (#3798) 2025-04-27 11:48:15 +08:00
CMakeLists.txt fix partialMatch (#3413) 2025-04-11 16:42:52 +08:00
guidedDecoderTest.cpp [TRTLLM-4460] test: Use Llama 3.2 1B for Llama C++ tests (#3206) 2025-05-01 05:31:08 +08:00
modelSpec.cpp Update TensorRT-LLM (#2873) 2025-03-11 21:13:42 +08:00
modelSpec.h Update TensorRT-LLM (#2873) 2025-03-11 21:13:42 +08:00
modelSpecBinding.cpp Update TensorRT-LLM (#2873) 2025-03-11 21:13:42 +08:00
peftCacheManagerTest.cpp Update TensorRT-LLM (#2873) 2025-03-11 21:13:42 +08:00
trtEncoderModelTest.cpp Update TensorRT-LLM (#2873) 2025-03-11 21:13:42 +08:00
trtGptModelRealDecoderTest.cpp [TRTLLM-4460] test: Use Llama 3.2 1B for Llama C++ tests (#3206) 2025-05-01 05:31:08 +08:00
trtGptModelTest.cpp feat: Add multimodal embedding field in LlmRequest (#3855) 2025-05-01 12:23:30 +08:00