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
..
_torch	refactor: remove ParallelConfig in tensorrt_llm._torch.distributed module (#3370)	2025-04-11 15:34:20 -07:00
api_stability	chore: refactor the LlmArgs with Pydantic and migrate remaining pybinding configs to python (#3025)	2025-04-05 13:31:48 +08:00
bindings	feat: Add numNodes to ParallelConfig (#3346)	2025-04-13 13:55:04 +02:00
llmapi	test: fix memory leak of tests (#3392)	2025-04-10 14:31:40 +08:00
others	test: reorganize tests folder hierarchy (#2996)	2025-03-27 12:07:53 +08:00
scaffolding	feat: Make scaffolding Controller more generic #3408 (#3416)	2025-04-12 21:35:38 +08:00
tools	Update TensorRT-LLM (#2936)	2025-03-18 21:25:19 +08:00
trt	Waive unittest/trt/model/test_mamba.py::TestMamba::test_loaders_mamba_130m_hf_from_checkpoint. Will fix it later. (#3356)	2025-04-07 22:36:35 -07:00
utils	test: reorganize tests folder hierarchy (#2996)	2025-03-27 12:07:53 +08:00
conftest.py	Add thread leak check and fix thread/memory leak issues. (#3270)	2025-04-08 19:03:18 +08:00
dump_checkpoint_stats.py	Update TensorRT-LLM (#2936)	2025-03-18 21:25:19 +08:00
profile_utils.py	Update TensorRT-LLM (#2936)	2025-03-18 21:25:19 +08:00
pytest.ini	chore: Refine attention backend interface. (#3271)	2025-04-09 02:34:53 +08:00
test_model_runner_cpp.py	Update TensorRT-LLM (#2936)	2025-03-18 21:25:19 +08:00
test_pip_install.py	relax the limitation of setuptools (#2992)	2025-03-24 13:36:10 +08:00