supports internode_ll_two_stage (#4162)

* supports internode_ll_two_stage

* supports internode_ll_two_stage

* supports internode_ll_two_stage

* supports internode_ll_two_stage

* supports D internode_ll_two_stage

* fix codestype

* fix xpu internode_ll_two_stage

* fix xpu internode_ll_two_stage
This commit is contained in:
lzy
2025-11-04 16:35:40 +08:00
committed by GitHub
parent 8a40374bfe
commit af7e0f27f3
6 changed files with 165 additions and 38 deletions

View File

@@ -541,6 +541,8 @@ class ParallelConfig:
self.engine_pid: Optional[int] = None
# Do profile or not
self.do_profile: bool = False
# Use internode_ll_two_stage or not
self.use_internode_ll_two_stage: bool = False
self.pod_ip: str = None
# enable the custom all-reduce kernel and fall back to NCCL(dist.all_reduce).