i-robot
2fbec9a554
!27856 use neighbor-exchange-v2 for conv2d
...
Merge pull request !27856 from yangzhenzhang/use-neighborexchangev2-for-conv2d
2021-12-21 10:59:54 +00:00
yangzhenzhang
8a68577756
use neighbor-exchange-v2 for conv2d
2021-12-21 14:27:41 +08:00
wzw
a9b78682d5
parallel ut refactor 3
2021-12-21 14:26:58 +08:00
yangzhenzhang
5f6477b022
add output strategy for gather op
2021-12-16 09:24:50 +08:00
i-robot
d49f5e6caf
!27525 support optimizer parallel for adafactor
...
Merge pull request !27525 from yangzhenzhang/support-opt-parallel-for-adafactor
2021-12-15 01:25:12 +00:00
yao_yf
30576c6a75
fix reshape bool type in auto parallel
2021-12-13 20:24:33 +08:00
yangzhenzhang
2a0b528084
support opt parallel for adafactor
2021-12-13 11:21:44 +08:00
i-robot
938dc8abd0
!27439 [Auto parallel] Add new operatorInfo for Parallel: CumSum
...
Merge pull request !27439 from Xiaoda/117-add-cumsum-op
2021-12-10 07:00:06 +00:00
i-robot
0e358f4cb3
!27428 revert insert VirtualDataset node for master
...
Merge pull request !27428 from lilei/modify_virtualdataset_for_master
2021-12-09 02:47:13 +00:00
Xiaoda Zhang
8042c88223
add the new operatorInfo for parallel: CumSum
2021-12-09 09:21:52 +08:00
lilei
2edf6ab33b
revert insert VirtualDataset node for master
2021-12-08 17:39:42 +08:00
i-robot
faaec746f7
!27401 add more ut tests for allreduce fusion
...
Merge pull request !27401 from jiahongQian/master
2021-12-08 09:38:37 +00:00
jiahongQian
b03c8d18d3
add more ut tests
2021-12-08 14:17:02 +08:00
i-robot
ffca7b08a5
!27237 auto insert VirtualDataset node for master
...
Merge pull request !27237 from lilei/modify_virtualdataset_for_master
2021-12-07 09:02:50 +00:00
i-robot
f40668ef73
!27251 test_micro_batch_Interleaved
...
Merge pull request !27251 from lilei/add_parallel_ut
2021-12-07 02:18:09 +00:00
lilei
05189459ab
auto insert VirtualDataset node for master
2021-12-06 21:39:37 +08:00
lilei
e933aa268b
test_micro_batch_Interleaved
2021-12-06 18:37:21 +08:00
i-robot
2d23b698a6
!27024 add allreduce fusion by size
...
Merge pull request !27024 from jiahongQian/master
2021-12-06 05:32:12 +00:00
q00596439
de36fdc169
add allreduce fusion size and unify the interface
2021-12-05 21:35:11 +08:00
huangxinjing
8c9b2b93a8
Add transformer
2021-12-03 09:08:57 +08:00
yangzhenzhang
7454b8f8f2
check args for shard
2021-12-02 11:51:29 +08:00
Xiaoda Zhang
364858cbc9
In sharding propagation, to keep strategy consistent of parameter being used by multiple operators, we check the edge with one node of TmpIdentityInfo
2021-11-30 18:58:33 +08:00
Xiaoda Zhang
04db51a528
In a previous PR ( https://gitee.com/mindspore/mindspore/pulls/26807/ ), we replaced 'auto_parallel_search_mode' by 'search_mode' directly.
...
However, to be forward compatitable, it is suitable to keep 'auto_parallel_search_mode' available. This PR recovers the 'auto_parallel_search_mode' interface and adds a warning when using this old interface.
This PR also deals with other codestyle things.
2021-11-29 15:40:23 +08:00
i-robot
9f8ec2c5ab
!26807 [Auto parallel] [Sharding propagation] Interface change of sharding propagation
...
Merge pull request !26807 from Xiaoda/113-auto-parallel-search-mode-changes-to-search-mode
2021-11-26 01:48:58 +00:00
i-robot
6ecbc97fd6
!26804 virtual_dataset_avoid_auto_parallel
...
Merge pull request !26804 from yao_yf/virtual_dataset_avoid_auto_parallel
2021-11-26 01:48:53 +00:00
i-robot
b282414de7
!26619 arallel_ut_refactoring
...
Merge pull request !26619 from 王志伟/parallel_ut_refactoring1
2021-11-26 01:34:49 +00:00
Xiaoda Zhang
ad5ac77ae8
1) 'auto_parallel_search_mode' changes to 'search_mode';
...
2) 'sharding_propagation' moves to 'search_mode';
2021-11-25 19:38:21 +08:00
yao_yf
f29ce1fb60
virtual dataset avoid auto parallel
2021-11-25 18:47:35 +08:00
i-robot
519f14a909
!26006 slice recompute activation
...
Merge pull request !26006 from yao_yf/add_transformer_slice_activation_config
2021-11-24 06:14:28 +00:00
wzw
86c5ad20c8
parallel_ut_refactoring1
2021-11-24 09:41:52 +08:00
i-robot
1b8c2ff0e9
!26414 fault_recover_by_mirror_group_fix_opt_shard
...
Merge pull request !26414 from yao_yf/fault_recover_by_mirror_group_fix_opt_shard
2021-11-20 03:06:45 +00:00
yao_yf
188d39da83
slice_activation_in_recompute
...
slice recompute activation
2021-11-19 10:51:54 +08:00
yao_yf
01dc4bbdf9
fix fault recover in optimizer shard
2021-11-19 09:46:27 +08:00
Xiaoda Zhang
df67e74eaf
making sharding_propagation smooth, add a reshape justification:
...
1) when propagate sharding strategy from one op to another, try to find the strategy with zero communication cost;
2) if there is no such strategy, find the strategy with minimum communication cost, and raise a warning;
2021-11-18 15:55:50 +08:00
i-robot
9f52343a6a
!26350 add check for resizenearestneighbor parallel op
...
Merge pull request !26350 from yangzhenzhang/add-check-for-resize-op
2021-11-18 05:45:57 +00:00
yangzhenzhang
ba99e4c505
add check for resize op
2021-11-16 15:39:52 +08:00
ttudu
33ac1de062
fix bug
2021-11-16 11:14:58 +08:00
i-robot
7a73bae5c3
!26036 add output strategy for matmul operator
...
Merge pull request !26036 from yangzhenzhang/add-output-strategy-for-op-init
2021-11-12 01:22:22 +00:00
Xiaoda Zhang
a772767265
support reshape in sharding propagation:
...
1) using 'swc index of strategy_cost_' as reshape's selected strategy;
2) when encountering reshape in BFS, select the 'swc index' with zero communication cost;
3) when encountering a reshape that is already visited, check whether there exists communication between reshape and current operator. It is OK if communication happens between two configured operators;
4) currently, two consecutive reshapes are not supported;
5) adjusting BFS structure in graph_costmodel.cc;
6) adjusting some code in step_auto_parallel.cc to avoid cyclomatic complexity.
2021-11-11 11:35:43 +08:00
yangzhenzhang
8431ba616c
add output strategy for op init
2021-11-10 15:35:45 +08:00
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
i-robot
ded1c77bbf
!25765 neighborExchangeV2 & grad
...
Merge pull request !25765 from TuDouNi/neighborExchangeV2
2021-11-05 09:31:03 +00:00
ttudu
e953c15cd2
NeighborExchangeV2 & Grad
2021-11-05 11:00:46 +08:00
yangzhenzhang
6ad6304b77
add output strategy
2021-11-04 16:28:54 +08:00
yangzhenzhang
c42081619e
add parallel op for resizenearestneighbor
2021-10-28 11:31:00 +08:00
wanyiming
4fbc59a98a
utfixs
2021-10-18 18:06:37 +08:00
i-robot
f83070728d
!24790 support user define strategy gen method under auto parallel context
...
Merge pull request !24790 from zhuyuxiao/master
2021-10-14 09:03:36 +00:00
zhuyuxiao
1907246931
change api
2021-10-13 10:28:30 +08:00
i-robot
3fd94000c5
!24568 Apply batch parallel in auto_parallel mode when strategies are not specified
...
Merge pull request !24568 from zhuyuxiao/master
2021-10-13 01:20:15 +00:00
zhuyuxiao
cf76c76745
apply batch parallel in auto_parallel mode when strategies are not specified
2021-10-12 15:44:10 +08:00