mindspore/tests/st/auto_parallel
caifubi d490450270 PyNative Bprop run in Graph.
1. Dynamic structure bp graph need to be split and run op by op.
2. Static bp graph can run graph integated. (Only support
CPU/GPU/Non-DynamaiShape)
2022-05-27 23:26:46 +08:00
..
__init__.py initial version 2020-03-27 22:54:54 +08:00
cell_shard.py PyNative Bprop run in Graph. 2022-05-27 23:26:46 +08:00
env.sh add use case 2020-11-30 10:39:29 +08:00
multifieldembeddinglookup_parallel.py Add bprop_return_sparse flag for sparse bprop primitive and remove context option: 'enable_sparse'. 2022-05-19 17:31:57 +08:00
onehot_model_parallel.py change API set_strategy() to shard() 2020-09-10 15:30:19 +08:00
optimizer_parallel.py modify if xxx is True to if xxx and modify if xxx is not True to if not xxx 2022-01-12 22:31:49 +08:00
parallel_strategy_search.py modify if xxx is True to if xxx and modify if xxx is not True to if not xxx 2022-01-12 22:31:49 +08:00
resnet50_expand_loss.py Change Loss to LossBase 2021-07-02 16:41:02 +08:00
run_auto_parallel_loss_expand.sh fix auto parallel st 2020-03-31 11:19:06 +08:00
run_auto_parallel_resnet50_expand_loss.sh add auto parallel st 2020-05-29 10:53:50 +08:00
run_cell_shard.sh add st for shard 2022-03-01 15:32:37 +08:00
run_onehot_model_parallel.sh fix auto parallel st 2020-03-31 11:19:06 +08:00
run_optimizer_parallel.sh add optimizer_parallel level0 case 2020-12-17 18:55:36 +08:00
run_parallel_recursive_strategy_search.sh add auto parallel recursive use case 2020-12-14 17:13:10 +08:00
run_parallel_strategy_search.sh add auto parallel recursive use case 2020-12-14 17:13:10 +08:00
soft_entropy_loss_expand_parallel.py change API set_strategy() to shard() 2020-09-10 15:30:19 +08:00
test_cell_shard.py add st for shard 2022-03-01 15:32:37 +08:00
test_expand_loss.py optim pylint 2020-08-31 14:49:57 +08:00
test_model_parallel_onehot.py fix pylint warnings 2020-05-22 14:54:36 +08:00
test_multifieldembeddinglookup_parallel.py st multi-hot 2021-01-30 10:44:20 +08:00
test_optimizer_parallel.py upgrade Ascend software pacakage 03 Aug 21 2021-08-03 19:24:18 +08:00
test_parallel_recursive_strategy_search.py open 8p testcase 2021-03-03 21:40:02 +08:00
test_parallel_strategy_search.py upgrade Ascend software pacakage 03 Aug 21 2021-08-03 19:24:18 +08:00
test_resnet50_expand_loss.py optim pylint 2020-08-31 14:49:57 +08:00