mindspore2022/tests/ut/python/parallel
i-robot 77424eaad5 !23004 Add args Check for Transformer
Merge pull request !23004 from huangxinjing/args_check
2021-09-08 08:24:06 +00:00
..
__init__.py set parallel communication init flag in parallel ut 2021-03-30 11:21:57 +08:00
conftest.py clean pylint 2020-05-18 16:42:35 +08:00
test_adafactor.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_add_relu_redistribution.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_allreduce_fusion.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_alltoall.py Revert "fix same node is used by two comm op" 2021-08-30 16:54:38 +08:00
test_arithmetic.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_BN_PReLU.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_activation.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_arithmetic.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_assign_sub_with_ref_key.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_cast.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_common_parameter.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_double_sources.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_double_star.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_double_subgraphs.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_fc_nobias.py Optimize the Executors routines. 2021-08-28 17:36:12 +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 Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_for_loop_multi_subgraph.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_for_loop_reshape.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_for_loop_simplify.py fix parallel timeout 2021-04-02 20:19:26 +08:00
test_auto_parallel_four_matmul.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_inference.py add notice 2020-06-29 17:39:25 +08:00
test_auto_parallel_l2normalize.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_matmul_drop.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_matmul_prelu.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_multi_graph.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_onehot.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_parameter_cast.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_partial_strategy.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_reduce_method.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_reshape.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_resnet.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_resnet_predict.py set device id master 0813 2021-09-02 20:51:50 +08:00
test_auto_parallel_resnet_sharding_propagation.py !21776 set device_id master 0813 2021-09-06 03:02:58 +00:00
test_auto_parallel_resnet_sharding_propagation2.py !21776 set device_id master 0813 2021-09-06 03:02:58 +00:00
test_auto_parallel_rhombus.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_segment_min.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_segment_sum.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_shard_propagation.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_shard_propagation2.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_shard_propagation3.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_softmax_loss.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_star_partial_strategy.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_transformer.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_transpose.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_triangle_overwrite.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_tuple_depend.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_two_bn.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_two_matmul.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_two_partial_matmul.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_parallel_zig_zag.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_auto_star_elimination.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_batch_matmul.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_batch_parallel.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_batch_parallel_dropout.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_batch_parallel_tensoradd.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_batchmm.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_batchnorm.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_batchnorm_batch_parallel.py modify batch_normal 2021-01-13 20:59:46 +08:00
test_bias_add.py add parallel operator for conv2d 2021-06-10 14:58:51 +08:00
test_bool_grad.py change API set_strategy() to shard() 2020-09-10 15:30:19 +08:00
test_broadcast_dict.py parallel envs variable check 2021-05-07 09:12:53 +08:00
test_broadcast_to.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_combined_tensor.py clean pylint 2020-05-18 16:42:35 +08:00
test_comm_not_recompute.py recompute interface modify 2021-09-01 10:59:59 +08:00
test_comparison_function_info.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_concat.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_control_flow.py auto_parallel_support_control_flow 2021-07-22 17:31:31 +08:00
test_conv2d.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_conv2d_transpose.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_cus_matmul_dds.py parallel_sparse_attention_ops_fix_repeated_cal 2021-09-06 19:57:22 +08:00
test_dataset.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_dataset_interface.py [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 2021-08-22 16:26:45 +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 Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_different_type_for_div_op.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_distribute_predict.py remove restriction for opt shard in inference 2021-07-16 17:16:12 +08:00
test_dropout.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_dropout_do_mask.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_dsd_matmul.py add parallel sparse attention ops: dsd_matmul 2021-08-30 18:46:52 +08:00
test_dynamic_shape.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_element_wise_function.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_embeddinglookup.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_eval.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_expand_dims.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_forward_graph.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_full_batch.py Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 2021-08-23 01:46:38 +08:00
test_gather_v2.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_gather_v2_primitive.py [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 2021-08-22 16:26:45 +08:00
test_gatherd.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_gathernd.py add gathernd op 2021-04-16 14:29:03 +08:00
test_gathernd_further.py add parallel gathernd test case 2021-05-28 17:21:11 +08:00
test_get_next.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_get_parameter_layout.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_gpu_dropout.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_hybird_parallel_activation.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_initializer_weight_slice.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_l2normalize.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_layer_norm.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_layer_norm_further.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_linear.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_load_tensor.py clean pylint 2020-05-18 16:42:35 +08:00
test_loop_two_matmul.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_loss_and_o2_level.py fix parallel timeout 2021-04-02 20:19:26 +08:00
test_loss_and_optimizer.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_loss_scale.py [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 2021-08-22 16:26:45 +08:00
test_manual_embedding_lookup.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_manual_gatherv2.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_matmul_dropout.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_matmul_tensor.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_maxpool_avgpool.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_mix_precision_hybrid_parallel.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_model_with_loss.py fix parallel timeout 2021-04-02 20:19:26 +08:00
test_model_without_loss.py fix parallel timeout 2021-04-02 20:19:26 +08:00
test_mul_div_bn.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_multi_field_embedding.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_neg.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_neighborexchange.py Revert "fix same node is used by two comm op" 2021-08-30 16:54:38 +08:00
test_o2_level.py fix parallel timeout 2021-04-02 20:19:26 +08:00
test_one_dev.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_one_hot_net.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_one_weight_parameter.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_onehot.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_onehot_2dim.py Optimize the Executors routines. 2021-08-28 17:36:12 +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 Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_optimizer_clone_weight.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_pack.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_parallel_optimizer.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_parallel_transformer.py Add args check 2021-09-08 09:12:35 +08:00
test_parameter_init.py change API set_strategy() to shard() 2020-09-10 15:30:19 +08:00
test_parameter_merge.py fix parallel timeout 2021-04-02 20:19:26 +08:00
test_parameter_multi_users.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_pipeline_shared_parameter.py Refactor_part_of_pipeline 2021-08-24 21:37:41 +08:00
test_pipeline_split.py pipeline_support_predict_master 2021-06-26 09:25:59 +08:00
test_prelu.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_prelu_cell.py change API set_strategy() to shard() 2020-09-10 15:30:19 +08:00
test_print.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_range.py fix parallel timeout 2021-04-02 20:19:26 +08:00
test_reduce_method_info.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_reluv2.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_repeated_calc.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_reshape.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_reshape_optimized.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_reshape_parameter.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_reshape_skip_redistribution.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_reshape_unexpand.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_scalar_loss.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_scatter_update.py modify check strategy for scatter update 2021-05-10 11:22:00 +08:00
test_select.py Making the Tile operator to have more parallel strategies 2021-04-27 11:36:40 +08:00
test_self_attention.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_semi_auto_two_subgraphs.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_set_auto_parallel_context.py fix optimizer weight shard config 2021-06-08 10:44:17 +08:00
test_shared_param_and_mix_precision.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_sigmoid_cross_entropy_with_logits.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_slice.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_softmax_cross_entropy_loss.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_sparse_feature_bprop.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_sparse_gather_v2.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_split.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_split_grad_sens.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_square.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_squeeze_info.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_step_parallel.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_strategy_checkpoint.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_stridedslice.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_sum_as_loss.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_tile.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_topk.py add topk op 2021-04-17 14:12:49 +08:00
test_train_and_eval.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_transpose.py change API set_strategy() to shard() 2020-09-10 15:30:19 +08:00
test_two_matmul.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_two_weights_parameter.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_uniform_candidate_sampler.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_unsortedsegmentmax.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_unsortedsegmentmin.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_unsortedsegmentsum.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_using_seed_for_initializer.py modify_normal 2020-10-22 21:35:44 +08:00
test_virtual_dataset_3_input.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_virtual_dataset_with_strategy.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_virtual_output.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00