m53297601
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
1089 Commits
11 Branches
11 Tags
1004 MiB
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from 'bbdfd92979'
${ noResults }
Commit Graph

12 Commits (bbdfd92979e2ff209d740ece2920fb5e3349ffbb)

Author SHA1 Message Date
ch-l f806b72447 use DeviceMemory for memory control
6 years ago
yangzhenzhang 6d522f0a4f add parallel op for layernorm
6 years ago
Xiaoda Zhang 0ac50a19f5 Model the memory cost in auto-parallel. It is calculated by the output of operators, plus the parameters. Additionally, modify the graph-operations in auto_parallel to include memory_cost.
6 years ago
c00425699 d62f560b50 add_bool_type_check_in_comm_op
6 years ago
buxue 5841fe010e Support pow's second input could be tensor and fix bug in bprop of pow
6 years ago
yangzhenzhang b34c0e7a17 add parallel op for dropoutdomask
6 years ago
c00425699 b413638f23 refactor OperatorCostPtr in OperatorInfo
6 years ago
mindspore-ci-bot 2e6e94b2b6 !177 prelu operator support parallel on the channel
6 years ago
yao_yf b5e3fa9593 fix auto parallel prelu
6 years ago
Xiaoda Zhang a153fad874 This commit is to separate the computation cost and memory cost in auto_parallel. Some related memory correction is removed.
6 years ago
c00425699 3bb48ffee1 use std::vector instead of std::list to promote performance for parallel module
6 years ago
zhunaipan 930a1fb0a8 initial version
6 years ago