TensorRT-LLMs/tensorrt_llm/auto_parallel
Yanchao Lu b4b1185af3
[https://nvbugs/5450855][fix] Cherry pick #6700 and #6702 from main (#6808)
Signed-off-by: Yiqing Yan <yiqingy@nvidia.com>
Signed-off-by: Yanchao Lu <yanchaol@nvidia.com>
Co-authored-by: Yiqing Yan <yiqingy@nvidia.com>
2025-08-12 18:11:47 +08:00
..
tensor_parallel
__init__.py
auto_parallel.py Release 0.20 to main (#4577) 2025-05-28 16:25:33 +08:00
cluster_info.py [https://nvbugs/5450855][fix] Cherry pick #6700 and #6702 from main (#6808) 2025-08-12 18:11:47 +08:00
config.py
device_mesh.py
node_graph.py
parallelization.py
pipeline_graph.py
runtime_profiling.py
shape_info.py
simplifier.py
solver.py
utils.py