huangxinjing
d777742904
1. Move the class to mindspore.parallel, support activation sharding
2021-08-20 14:34:10 +08:00
i-robot
dde05c36b8
!21551 auto_parallel_context interface dataset_strategy union
...
Merge pull request !21551 from yao_yf/auto_parallel_context_dataset_strategy_union
2021-08-18 09:03:18 +00:00
i-robot
415275ae17
!21805 support adafactor model parallel
...
Merge pull request !21805 from yangzhenzhang/auto-parallel-support-adafactor-opt
2021-08-18 07:20:39 +00:00
i-robot
0d839fa7c6
!21809 Improved Transformer Struture and Add Args Check
...
Merge pull request !21809 from huangxinjing/transformer_improved
2021-08-17 12:05:25 +00:00
i-robot
a77a0b968d
!21761 comm_recompute_interface.
...
Merge pull request !21761 from yao_yf/comm_recompute_interface
2021-08-17 09:14:29 +00:00
yangzhenzhang
7ca64d2235
auto parallel support adafactor opt
2021-08-17 10:36:42 +08:00
yao_yf
5277b229be
add cell comm recompute interface
2021-08-17 09:23:21 +08:00
huangxinjing
18044aff0f
1. Add docstring, elimitate attention mask, tuple append the deocoder return layer past
...
split attention inputs from two into three
Add input check and modify test case
Fix check error and add check to transformer inputs
add input check to the transformers
2021-08-16 21:08:48 +08:00
yao_yf
a83bf73298
union auto_parallel_context interface dataset_strategy
2021-08-16 10:54:46 +08:00
yao_yf
e233880e41
fix reshape depend reshape in auto parallel
2021-08-12 11:45:28 +08:00
i-robot
63445ff6fd
!21627 alltoall exception handle
...
Merge pull request !21627 from zhoufeng/alltoall_exception
2021-08-12 01:40:22 +00:00
yangzhenzhang
d18c813ee4
check strategy for conv2d
2021-08-11 19:12:48 +08:00
zhoufeng
03a56f2bb0
alltoall exception handle
...
Signed-off-by: zhoufeng <zhoufeng54@huawei.com>
2021-08-11 17:30:49 +08:00
i-robot
4aaa8126a0
!21528 Add Parallel Print Support
...
Merge pull request !21528 from huangxinjing/add_print
2021-08-09 10:55:04 +00:00
huangxinjing
92bad162bd
Add print
2021-08-09 11:14:25 +08:00
huangbingjian
53b31abf12
remove useless depend
2021-08-05 20:21:24 +08:00
yangzhenzhang
ef0361a449
fix bugs for conv2d
2021-08-03 10:48:05 +08:00
Xiaoda Zhang
4b4b3cdaf4
add reduceany operator and extend onehot to multi-dimensions
2021-08-02 08:53:43 +08:00
huangxinjing
615d1a179d
Add transformer layer
2021-07-29 11:10:43 +08:00
i-robot
9f296c58d6
!20960 [AutoParallel]Add replace graph for conv2d
...
Merge pull request !20960 from lichen/add_replace_graph_for_conv2d
2021-07-28 07:06:41 +00:00
lichenever
a7f8024c29
add_replace_graph_for_conv2d
2021-07-28 11:02:21 +08:00
yao_yf
dc7dc7d3fa
dataset strategy set
2021-07-27 18:44:10 +08:00
yangzhenzhang
80e5cc0e52
add parallel op for gatherd
2021-07-26 11:14:25 +08:00
Xiaoda Zhang
bb5d4212f7
enable All2All in infering redistribution ops
2021-07-24 09:09:01 +08:00
lichenever
3c7cfb7c08
auto_parallel_support_control_flow
2021-07-22 17:31:31 +08:00
i-robot
a7d40fc220
!20520 [AutoParallel]Add op AllToAllv
...
Merge pull request !20520 from lichen/add_op_AllToAllv
2021-07-20 07:57:03 +00:00
lichenever
8c1998fd6b
add_op_AllToAllv
2021-07-20 10:49:28 +08:00
i-robot
c9d3c1d346
!20411 enable optimizer parallel for inference
...
Merge pull request !20411 from gziyan/enable_opt_shard_predict
2021-07-20 01:37:06 +00:00
yangzhenzhang
b31cd27a08
update check strategy for conv2d
2021-07-19 15:03:26 +08:00
Ziyan
1c9166e0a6
remove restriction for opt shard in inference
2021-07-16 17:16:12 +08:00
Xiaoda Zhang
04381273b3
Add the sharding propagation function:
...
1) users configure sharding strategies for operators;
2) framework will propagate the strategies from configured-ops to
non-configured ops using BFS;
3) the propagation goal is to minimize redistribution communication
cost;
2021-07-14 10:54:18 +08:00
chenhaozhe
086a871975
Change Loss to LossBase
2021-07-02 16:41:02 +08:00
lichenever
db8850a4a3
pipeline_support_predict_master
2021-06-26 09:25:59 +08:00
Ziyan
be1f5a43d7
opt shard fit micro batch
2021-06-24 14:52:19 +08:00
yangzhenzhang
69acf757d0
add parallel op for conv2d backprop input
2021-06-24 10:01:49 +08:00
yangzhenzhang
24370b5613
add parallel op for maxpool
2021-06-21 09:00:56 +08:00
yangzhenzhang
af0d28de48
add parallel op for batchnorm
2021-06-17 15:43:24 +08:00
i-robot
85d860e6a2
!16457 [AutoParallel]pipeline_split_adapt_master
...
Merge pull request !16457 from lichen/pipeline_split_adapt_master
2021-06-11 11:37:40 +08:00
lichenever
db5d508356
pipeline_split_adapt_master
2021-06-10 20:17:33 +08:00
yangzhenzhang
7a40741048
add parallel operator for conv2d
2021-06-10 14:58:51 +08:00
Ziyan
95ac0f6d58
fix optimizer weight shard config
2021-06-08 10:44:17 +08:00
chenhaozhe
9da8534396
change _Loss to Loss
2021-06-03 15:26:59 +08:00
mindspore-ci-bot
1c8fda25ef
!16478 handle load op in step parallel
...
From: @gong_zi_yan
Reviewed-by: @yangzhenzhang,@stsuteng
Signed-off-by: @stsuteng
2021-05-29 09:32:35 +08:00
mindspore-ci-bot
b45b63fc58
!17239 add parallel gathernd test case
...
From: @hanyang001
Reviewed-by: @stsuteng,@yangzhenzhang
Signed-off-by: @stsuteng
2021-05-29 09:11:25 +08:00
Wan Hanyang
c51dff2634
add parallel gathernd test case
2021-05-28 17:21:11 +08:00
Wan Hanyang
3ce521d78f
add parallel layernorm test case
2021-05-28 17:20:04 +08:00
Ziyan
4b17493e52
handle load in step parallel
2021-05-28 09:16:40 +08:00
yangzhenzhang
d711d98f07
clean duplicate code
2021-05-25 17:25:34 +08:00
yao_yf
732d13ccff
parallel dropout support repeated compute
2021-05-20 19:42:13 +08:00
yangzhenzhang
6aa3859131
modify check strategy for scatter update
2021-05-10 11:22:00 +08:00