Wenhao Chen 3c08f17348 [hotfix]: modify create_ep_hierarchical_group and add test (#5032) ...

* feat: modify create_ep_hierarchical_group args

* test: add ep tests

* fix: remove get_process_group_ranks

* fix: fix src_rank

2023-11-17 10:53:00 +08:00

..

d_tensor

[gemini] gemini support tensor parallelism. (#4942)

2023-11-10 10:15:16 +08:00

moe_tensor

[hotfix]: modify create_ep_hierarchical_group and add test (#5032)

2023-11-17 10:53:00 +08:00

__init__.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

colo_parameter.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

colo_tensor.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

comm_spec.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

param_op_hook.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

shape_consistency.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

sharding_spec.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

utils.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00