kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-14 06:27:45 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity
75e13f4f88
TensorRT-LLMs/cpp/include/tensorrt_llm
History
Aurelien Chartier 7b38018fa0
feat: Add numNodes to ParallelConfig (#3346) 
* Add numNodes to ParallelConfig

If not provided, attempt to find the number of nodes by
adding the number of local ranks 0

Update device IDs check accordingly

Signed-off-by: Aurelien Chartier <achartier@nvidia.com>

* Add ParallelConfig pickle test

Signed-off-by: Aurelien Chartier <achartier@nvidia.com>

---------

Signed-off-by: Aurelien Chartier <achartier@nvidia.com>
2025-04-13 13:55:04 +02:00
..
batch_manager refactor: batch slot management in decoder classes (#3300) 2025-04-13 05:05:13 +08:00
common chore: Stabilize ABI boundary for internal kernel library (#3117) 2025-04-11 15:07:50 +08:00
deep_gemm fix: remove DeepGEMM line info (#3411) 2025-04-09 18:01:02 +08:00
executor feat: Add numNodes to ParallelConfig (#3346) 2025-04-13 13:55:04 +02:00
kernels Update TensorRT-LLM (#2873) 2025-03-11 21:13:42 +08:00
layers v1.2 (#3082) 2025-03-26 23:31:29 +08:00
plugins/api Update TensorRT-LLM (#2532) 2024-12-04 21:16:56 +08:00
runtime feat: Add numNodes to ParallelConfig (#3346) 2025-04-13 13:55:04 +02:00