mindspore2022/mindspore/parallel
huangxinjing f354ab22a3 add pipeline shard interface
Add support for no pipeline accugradient

Add delay tag for fusion op

Optimizer the visite order

add mirror for mini step control

Move the group to attributes

Add gradient_shard control for the mini step

Fix code stype

Fix ut description

Add interface
2021-11-09 09:11:13 +08:00
..
mpi Fix codex for python file 2021-05-21 08:57:35 +08:00
nn clean code 2021-10-30 09:23:23 +08:00
OWNERS add commiter yao_yf 2021-08-23 19:50:10 +08:00
__init__.py refine the comments of parallel-related interfaces 2021-10-12 19:35:24 +08:00
_auto_parallel_context.py add pipeline shard interface 2021-11-09 09:11:13 +08:00
_cell_wrapper.py fix auto optimizer weight shard 2020-11-02 16:25:48 +08:00
_cost_model_context.py modify code check for master 2021-08-16 15:58:15 +08:00
_dp_allreduce_fusion.py Add code check 2021-09-14 19:43:51 +08:00
_ps_context.py fixed enable ssl 2021-09-06 09:43:31 +08:00
_tensor.py add tensor.ndim and rename tensor.size() to tensor.size 2020-12-18 17:15:42 +08:00
_utils.py add pipeline shard interface 2021-11-09 09:11:13 +08:00
algo_parameter_config.py adjusting symbols 2021-11-05 10:02:12 +08:00