mindspore/tests/ut/python/parallel
huangbingjian c17a7ffbbd Replace ms_function with jit 2022-10-29 11:32:53 +08:00
..
utils rm set auto parallel 2022-10-28 09:25:22 +08:00
__init__.py add security isolate for save_graphs 2021-09-14 10:24:12 +08:00
conftest.py add security isolate for save_graphs 2021-09-14 10:24:12 +08:00
test_adafactor.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_add_relu_redistribution.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_addn.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_allreduce_fusion.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_alltoall.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_arithmetic.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_BN_PReLU.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_activation.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_adasum.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_arithmetic.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_assign_sub_with_ref_key.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_cast.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_common_parameter.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_data_parallel_mode.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_double_sources.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_double_star.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_double_subgraphs.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_fc_nobias.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_flag.py delete auto parallel flag 2022-07-30 15:19:56 +08:00
test_auto_parallel_for_loop.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_for_loop_multi_subgraph.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_for_loop_reshape.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_for_loop_simplify.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_auto_parallel_four_matmul.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_inference.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_l2normalize.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_matmul_drop.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_matmul_prelu.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_multi_graph.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_onehot.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_pangu_alpha_shard_propagation.py Avoid subgraphs returning parameter Ref type data by inserting Load, preventing run graph mode with kernelbykernel. 2022-10-11 18:56:16 +08:00
test_auto_parallel_parameter_cast.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_partial_strategy.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_reduce_method.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_reshape.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_resnet.py remove usage of ScalarToArray 2022-10-13 14:34:06 +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 remove usage of ScalarToArray 2022-10-13 14:34:06 +08:00
test_auto_parallel_resnet_sharding_propagation2.py remove usage of ScalarToArray 2022-10-13 14:34:06 +08:00
test_auto_parallel_rhombus.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_segment_min.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_segment_sum.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_shard_propagation.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_shard_propagation2.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_shard_propagation3.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_softmax_loss.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_star_partial_strategy.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_transformer.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_transpose.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_triangle_overwrite.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_tuple_depend.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_two_bn.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_two_matmul.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_two_partial_matmul.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_parallel_zig_zag.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_auto_star_elimination.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_autoparallel_rec_shared_param_strmodif.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_batch_matmul.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_batch_parallel.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_batch_parallel_dropout.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_batch_parallel_tensoradd.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_batchmm.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_batchnorm.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_batchnorm_batch_parallel.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_batchparallel_replace_shape.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_bias_add.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_bool_grad.py change API set_strategy() to shard() 2020-09-10 15:30:19 +08:00
test_bounding_box_encode.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_broadcast_dict.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_broadcast_to.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_cdist.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_cell_shard_check.py Add nested usage of shard verification 2022-10-26 08:00:46 +08:00
test_checkvalid.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_combined_tensor.py clean pylint 2020-05-18 16:42:35 +08:00
test_comm_fusion.py remove_parallel_related_in_model 2022-10-17 10:01:17 +08:00
test_comm_not_recompute.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_comparison_function_info.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_concat.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_conformer.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_control_flow.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_conv2d.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_conv2d_transpose.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_conv3d.py add padlist check 2022-10-08 17:22:48 +08:00
test_crop_and_resize.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_cus_matmul_dds.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_dataset.py Optimize the Executors routines. 2021-08-28 17:36:12 +08:00
test_dataset_interface.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_dense_matmul.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_device_num_not_power_of_2.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_different_type_for_div_op.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_distribute_predict.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_dropout.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_dropout_do_mask.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_dsd_matmul.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_dynamic_shape.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_element_wise_function.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_embeddinglookup.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_eval.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_expand_dims.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_forward_graph.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_full_batch.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_func_mode.py Replace ms_function with jit 2022-10-29 11:32:53 +08:00
test_gather_field_split.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_gather_v2.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_gatherd.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_gathernd.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_gathernd_further.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_get_next.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_get_parameter_layout.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_global_norm.py 1. The main gol: Fix mixing inserting the AllReduce when where is no mirror appeared 2022-03-15 15:54:06 +08:00
test_gpu_dropout.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_hybird_parallel_activation.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_initializer_weight_slice.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_inplace_op.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_invert_populationcount.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_iou.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_kldiv_loss.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_l2_loss.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_l2normalize.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_layer_norm.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_layer_norm_further.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_lerp.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_lin_space.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_linear.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_load_tensor.py Revert "add value print for Parameter" 2022-10-11 09:54:02 +08:00
test_loop_two_matmul.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_loss_and_o2_level.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_loss_and_optimizer.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_loss_scale.py Avoid subgraphs returning parameter Ref type data by inserting Load, preventing run graph mode with kernelbykernel. 2022-10-11 18:56:16 +08:00
test_manual_embedding_lookup.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_manual_gatherv2.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_matmul_dropout.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_matmul_tensor.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_maxpool_avgpool.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_micro_batch_interleaved.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_mirror_insert.py fix issue of insert mirror;optimaze sharding propagation 2022-10-12 16:11:26 +08:00
test_mix_precision_hybrid_parallel.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_model_with_loss.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_model_without_loss.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_mul_div_bn.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_mul_softmax_net.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_multi_field_embedding.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_neg.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_neighborexchange.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_neighborexchangev2.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_o2_level.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_one_dev.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_one_hot_net.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_one_weight_parameter.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_onehot.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_onehot_2dim.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_operator_model_parallel.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_optimizer.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_optimizer_clone_weight.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_pack.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_parallel_cumprod.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_parallel_cumsum.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_parallel_mirror_group.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_parallel_moe.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_parallel_optimizer.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_parallel_optimizer_shared_cast.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_parallel_transformer.py Impove the transforer apis 2022-09-23 21:57:58 +08:00
test_param_broadcast_cell.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_parameter_init.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_parameter_merge.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_parameter_multi_users.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_pipeline_end_node.py optimizer_pipeline_parallel 2022-08-17 17:40:26 +08:00
test_pipeline_opt_detection.py remove_parallel_related_in_model 2022-10-17 10:01:17 +08:00
test_pipeline_remove_node.py remove_parallel_related_in_model 2022-10-17 10:01:17 +08:00
test_pipeline_shared_parameter.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_pipeline_split.py !42506 mv ge pass from frontend to backend 2022-10-17 07:57:50 +00:00
test_prelu.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_prelu_cell.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_print.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_random_choice_with_mask.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_range.py Fix bprop bugs of Ops Range 2022-09-14 18:17:59 +08:00
test_reduce_method_info.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_reluv2.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_repeated_calc.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_reshape.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_reshape_optimized.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_reshape_parameter.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_reshape_shard_propagation.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_reshape_skip_redistribution.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_reshape_unexpand.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_resizebilinear.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_roi_align.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_scalar_loss.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_scatter_update.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_select.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_self_attention.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_semi_auto_two_subgraphs.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_set_auto_parallel_context.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_sharding_propagation_for_conv2ds.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_shared_param_and_mix_precision.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_sigmoid_cross_entropy_with_logits.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_slice.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_softmax_cross_entropy_loss.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_softmax_gather_net.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_sparse_feature_bprop.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_sparse_gather_v2.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_split.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_split_grad_sens.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_splitv.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_square.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_squeeze_info.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_step_parallel.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_strategy_checkpoint.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_stridedslice.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_sum_as_loss.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_tile.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_topk.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_train_and_eval.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_transform_tensor.py add checkpoints transform api 2022-09-14 09:25:39 +08:00
test_transpose.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_two_matmul.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_two_weights_parameter.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_uniform_candidate_sampler.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_uniform_real.py auto parallel dataset handle in feed mode 2022-08-15 14:41:16 +08:00
test_unsortedsegmentmax.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_unsortedsegmentmin.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_unsortedsegmentprod.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_unsortedsegmentsum.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_using_seed_for_initializer.py modify_normal 2020-10-22 21:35:44 +08:00
test_virtual_dataset_3_input.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_virtual_dataset_with_strategy.py rm set auto parallel 2022-10-28 09:25:22 +08:00
test_virtual_output.py rm set auto parallel 2022-10-28 09:25:22 +08:00