mindspore2022/tests/ut/python/parallel
yangzhenzhang a70d616841 mini step grad accumulation 2021-03-01 10:20:12 +08:00
..
__init__.py fix pylint warnings 2020-05-22 14:54:36 +08:00
conftest.py clean pylint 2020-05-18 16:42:35 +08:00
test_add_relu_redistribution.py Change TensorAdd to Add, from r1.1 to master 2021-02-01 17:53:52 +08:00
test_allreduce_fusion.py [auto-monad] Support side-effects by auto-monad 2021-02-08 09:01:15 +08:00
test_alltoall.py refactor get cnode strategy 2020-10-24 15:06:26 +08:00
test_arithmetic.py Change TensorAdd to Add, from r1.1 to master 2021-02-01 17:53:52 +08:00
test_auto_parallel_BN_PReLU.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_auto_parallel_activation.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_auto_parallel_arithmetic.py set cnode's fullname when cloning 2020-12-04 15:59:44 +08:00
test_auto_parallel_assign_sub_with_ref_key.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_auto_parallel_cast.py move parallel-related black-list to core/ir, and fix the cloneCNode bug 2020-12-10 15:57:01 +08:00
test_auto_parallel_common_parameter.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_auto_parallel_double_sources.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_auto_parallel_double_star.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_auto_parallel_double_subgraphs.py use from_numpy and add do_copy option 2020-12-23 21:38:05 +08:00
test_auto_parallel_fc_nobias.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_auto_parallel_flag.py Change TensorAdd to Add, from r1.1 to master 2021-02-01 17:53:52 +08:00
test_auto_parallel_for_loop.py Change TensorAdd to Add, from r1.1 to master 2021-02-01 17:53:52 +08:00
test_auto_parallel_for_loop_multi_subgraph.py support for-loop in auto-parallel 2020-11-09 17:18:30 +08:00
test_auto_parallel_for_loop_reshape.py Change TensorAdd to Add, from r1.1 to master 2021-02-01 17:53:52 +08:00
test_auto_parallel_for_loop_simplify.py support for-loop in auto-parallel 2020-11-09 17:18:30 +08:00
test_auto_parallel_four_matmul.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_auto_parallel_inference.py add notice 2020-06-29 17:39:25 +08:00
test_auto_parallel_l2normalize.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_auto_parallel_matmul_drop.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_auto_parallel_matmul_prelu.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_auto_parallel_multi_graph.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_auto_parallel_onehot.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_auto_parallel_parameter_cast.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_auto_parallel_partial_strategy.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_auto_parallel_reduce_method.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_auto_parallel_reshape.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_auto_parallel_resnet.py Change TensorAdd to Add, from r1.1 to master 2021-02-01 17:53:52 +08:00
test_auto_parallel_resnet_predict.py support distributed predict 2020-12-07 16:12:45 +08:00
test_auto_parallel_rhombus.py Change TensorAdd to Add, from r1.1 to master 2021-02-01 17:53:52 +08:00
test_auto_parallel_segment_min.py Add UnsortedSegmentSum and UnosrtedSemenMin For Oparallel Implements 2020-10-28 08:57:02 +08:00
test_auto_parallel_segment_sum.py Add UnsortedSegmentSum and UnosrtedSemenMin For Oparallel Implements 2020-10-28 08:57:02 +08:00
test_auto_parallel_softmax_loss.py modify Gelu、FastGelu to GeLU and FastGeLU 2021-02-05 17:19:52 +08:00
test_auto_parallel_star_partial_strategy.py Change TensorAdd to Add, from r1.1 to master 2021-02-01 17:53:52 +08:00
test_auto_parallel_transformer.py Change TensorAdd to Add, from r1.1 to master 2021-02-01 17:53:52 +08:00
test_auto_parallel_transpose.py move parallel-related black-list to core/ir, and fix the cloneCNode bug 2020-12-10 15:57:01 +08:00
test_auto_parallel_triangle_overwrite.py Change TensorAdd to Add, from r1.1 to master 2021-02-01 17:53:52 +08:00
test_auto_parallel_tuple_depend.py modify Gelu、FastGelu to GeLU and FastGeLU 2021-02-05 17:19:52 +08:00
test_auto_parallel_two_bn.py Change TensorAdd to Add, from r1.1 to master 2021-02-01 17:53:52 +08:00
test_auto_parallel_two_matmul.py set cnode's fullname when cloning 2020-12-04 15:59:44 +08:00
test_auto_parallel_two_partial_matmul.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_auto_parallel_zig_zag.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_auto_star_elimination.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_batch_matmul.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_batch_parallel.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_batch_parallel_dropout.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_batch_parallel_tensoradd.py Change TensorAdd to Add, from r1.1 to master 2021-02-01 17:53:52 +08:00
test_batchnorm_batch_parallel.py modify batch_normal 2021-01-13 20:59:46 +08:00
test_batchnorm_ex_batch_parallel.py change dimension of input for FusedBatchNormEx from 2D to 4D in test_two_matmul_batchnorm_ex. 2021-02-25 09:37:59 +08:00
test_bias_add.py use from_numpy and add do_copy option 2020-12-23 21:38:05 +08:00
test_bn_prelu_cell.py change API set_strategy() to shard() 2020-09-10 15:30:19 +08:00
test_bool_grad.py change API set_strategy() to shard() 2020-09-10 15:30:19 +08:00
test_broadcast_dict.py fix pylint warnings 2020-05-22 14:54:36 +08:00
test_broadcast_to.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_combined_tensor.py clean pylint 2020-05-18 16:42:35 +08:00
test_comparison_function_info.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_concat.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_dataset.py modify Gelu、FastGelu to GeLU and FastGeLU 2021-02-05 17:19:52 +08:00
test_dataset_interface.py change API set_strategy() to shard() 2020-09-10 15:30:19 +08:00
test_dataset_util.py remove to_full_tensor and load_inputs in exexute stage 2020-08-14 11:40:27 +08:00
test_dense_matmul.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_different_type_for_div_op.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_distribute_predict.py fix infer rank list typo 2020-12-28 18:46:01 +08:00
test_dropout_do_mask.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_dynamic_shape.py Change GatherV2 to Gather r1.1 to master 2021-01-30 14:13:47 +08:00
test_element_wise_function.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_embeddinglookup.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_eval.py hange strategys of last nodes in eval/predict at auto parallel mode 2020-12-11 11:11:28 +08:00
test_expand_dims.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_forward_graph.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_full_batch.py raise RuntimeError when using full_batch neither under semi_auto_parallel nor auto_parallel 2020-09-16 14:51:09 +08:00
test_gather_v2.py Change GatherV2 to Gather r1.1 to master 2021-01-30 14:13:47 +08:00
test_gather_v2_primitive.py Change GatherV2 to Gather r1.1 to master 2021-01-30 14:13:47 +08:00
test_get_next.py change API set_strategy() to shard() 2020-09-10 15:30:19 +08:00
test_get_parameter_layout.py enable parallel optimizer in auto parallel 2020-10-08 14:51:15 +08:00
test_gpu_dropout.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_grad_accumulation.py mini step grad accumulation 2021-03-01 10:20:12 +08:00
test_hybird_parallel_activation.py modify Gelu、FastGelu to GeLU and FastGeLU 2021-02-05 17:19:52 +08:00
test_initializer_weight_slice.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_l2normalize.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_layer_norm.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_linear.py modify Gelu、FastGelu to GeLU and FastGeLU 2021-02-05 17:19:52 +08:00
test_load_tensor.py clean pylint 2020-05-18 16:42:35 +08:00
test_loop_two_matmul.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_loss_and_o2_level.py run cast before parallel optimizer 2020-11-13 14:04:04 +08:00
test_loss_and_optimizer.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_loss_scale.py [auto-monad] Support side-effects by auto-monad 2021-02-08 09:01:15 +08:00
test_manual_embedding_lookup.py [auto-monad] Support side-effects by auto-monad 2021-02-08 09:01:15 +08:00
test_manual_gatherv2.py Change GatherV2 to Gather r1.1 to master 2021-01-30 14:13:47 +08:00
test_matmul_dropout.py unify mindir 2020-12-12 17:55:16 +08:00
test_matmul_tensor.py Change TensorAdd to Add, from r1.1 to master 2021-02-01 17:53:52 +08:00
test_mix_precision_hybrid_parallel.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_model_with_loss.py add model with loss, without loso and o2 test case 2020-09-12 14:26:40 +08:00
test_model_without_loss.py add model with loss, without loso and o2 test case 2020-09-12 14:26:40 +08:00
test_mul_div_bn.py Change TensorAdd to Add, from r1.1 to master 2021-02-01 17:53:52 +08:00
test_multi_field_embedding.py [auto-monad] Support side-effects by auto-monad 2021-02-08 09:01:15 +08:00
test_neg.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_o2_level.py add model with loss, without loso and o2 test case 2020-09-12 14:26:40 +08:00
test_one_dev.py unify mindir 2020-12-12 17:55:16 +08:00
test_one_hot_net.py Change TensorAdd to Add, from r1.1 to master 2021-02-01 17:53:52 +08:00
test_one_weight_parameter.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_onehot.py modify Gelu、FastGelu to GeLU and FastGeLU 2021-02-05 17:19:52 +08:00
test_operator_model_parallel.py Change TensorAdd to Add, from r1.1 to master 2021-02-01 17:53:52 +08:00
test_optimizer.py rename mirror_mean to gradients_mean 2020-09-04 14:47:36 +08:00
test_optimizer_clone_weight.py change API set_strategy() to shard() 2020-09-10 15:30:19 +08:00
test_pack.py modify pack to stack 2021-02-04 18:54:31 +08:00
test_parallel_optimizer.py !9832 expose_allgather_fusion_to_users 2020-12-28 09:54:57 +08:00
test_parameter_init.py change API set_strategy() to shard() 2020-09-10 15:30:19 +08:00
test_parameter_merge.py merge parameter slice compile graph only once 2021-02-03 19:39:32 +08:00
test_parameter_multi_users.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_pipeline_split.py delete useless parameter in pipeline parallel 2021-01-21 19:05:40 +08:00
test_prelu.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_prelu_cell.py change API set_strategy() to shard() 2020-09-10 15:30:19 +08:00
test_range.py add range op 2020-11-09 15:14:07 +08:00
test_reduce_method_info.py Change TensorAdd to Add, from r1.1 to master 2021-02-01 17:53:52 +08:00
test_reluv2.py fix ReLUV2 error 2020-10-29 20:29:01 +08:00
test_repeated_calc.py Change TensorAdd to Add, from r1.1 to master 2021-02-01 17:53:52 +08:00
test_reshape.py Change TensorAdd to Add, from r1.1 to master 2021-02-01 17:53:52 +08:00
test_reshape_optimized.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_reshape_parameter.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_reshape_skip_redistribution.py Change GatherV2 to Gather r1.1 to master 2021-01-30 14:13:47 +08:00
test_reshape_unexpand.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_scalar_loss.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_self_attention.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_semi_auto_two_subgraphs.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_set_auto_parallel_context.py update pipeline parallel 2020-11-18 13:55:34 +08:00
test_sigmoid_cross_entropy_with_logits.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_slice.py Add slice parallel op 2020-11-18 20:20:33 +08:00
test_softmax_cross_entropy_loss.py modify Gelu、FastGelu to GeLU and FastGeLU 2021-02-05 17:19:52 +08:00
test_sparse_feature_bprop.py [auto-monad] Support side-effects by auto-monad 2021-02-08 09:01:15 +08:00
test_sparse_gather_v2.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_split.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_split_grad_sens.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_square.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_squeeze_info.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_step_parallel.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_strategy_checkpoint.py add group ckpt 2021-01-21 14:41:05 +08:00
test_stridedslice.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_sum_as_loss.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_tile.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_train_and_eval.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_transpose.py change API set_strategy() to shard() 2020-09-10 15:30:19 +08:00
test_two_matmul.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_two_weights_parameter.py set last node data parallel or repeat calculate in eval/predict 2020-10-26 20:05:00 +08:00
test_uniform_candidate_sampler.py Change GatherV2 to Gather r1.1 to master 2021-01-30 14:13:47 +08:00
test_unsortedsegmentmax.py Add UnsortedSegmentMax Operation 2020-11-19 17:41:54 +08:00
test_unsortedsegmentmin.py Add UnsortedSegmentMax Operation 2020-11-19 17:41:54 +08:00
test_unsortedsegmentsum.py Fix condition check 2020-12-18 16:43:43 +08:00
test_using_seed_for_initializer.py modify_normal 2020-10-22 21:35:44 +08:00
test_virtual_dataset_3_input.py modify Gelu、FastGelu to GeLU and FastGeLU 2021-02-05 17:19:52 +08:00