TensorRT-LLMs/tests/unittest/_torch/multi_gpu_modeling
Aurelien Chartier 3de82c41cd
Pytorch PP + attention DP support (#3044)
Signed-off-by: Aurelien Chartier <achartier@nvidia.com>
2025-03-28 00:11:19 +08:00
..
test_deepseek.py Pytorch PP + attention DP support (#3044) 2025-03-28 00:11:19 +08:00
test_llama.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00