mindspore/tests/ut/python/parallel
candanzg 2cc85bdc93 Support weight compile according to shape
Signed-off-by: candanzg <zhangshucheng@huawei.com>
2020-05-13 17:49:07 +08:00
..
parallel_end_to_end add parallel mode for cell 2020-05-07 20:11:20 +08:00
__init__.py implementing-searching-strategy-for-inference 2020-05-07 14:53:50 +08:00
conftest.py reset auto parallel context before each test 2020-05-12 14:59:45 +08:00
test_add_relu_redistribution.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_allreduce_fusion.py enable/disable allreduce_fusion 2020-04-24 10:34:59 +08:00
test_alltoall.py remove ge depend in cpu 2020-04-02 19:38:42 +08:00
test_arithmetic.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_auto_parallel_BN_PReLU.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_auto_parallel_arithmetic.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_auto_parallel_assign_sub_with_ref_key.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_auto_parallel_cast.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_auto_parallel_common_parameter.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_auto_parallel_double_star.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_auto_parallel_double_subgraphs.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_auto_parallel_fc_nobias.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_auto_parallel_four_matmul.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_auto_parallel_inference.py calculating PEAK memory cost in the inference phase 2020-05-12 15:30:59 +08:00
test_auto_parallel_l2normalize.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_auto_parallel_matmul_drop.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_auto_parallel_matmul_prelu.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_auto_parallel_onehot.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_auto_parallel_parameter_cast.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_auto_parallel_partial_strategy.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_auto_parallel_reduce_method.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_auto_parallel_reshape.py move InferStraByTensorInfo to tensor_info.h 2020-05-11 14:44:06 +08:00
test_auto_parallel_resnet.py remove two context param 2020-05-12 22:11:50 +08:00
test_auto_parallel_rhombus.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_auto_parallel_softmax_loss.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_auto_parallel_transformer.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_auto_parallel_transpose.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_auto_parallel_tuple_depend.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_auto_parallel_two_bn.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_auto_parallel_two_matmul.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_auto_parallel_two_partial_matmul.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_auto_parallel_zig_zag.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_auto_star_elimination.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_batch_matmul.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_batch_parallel.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_batch_parallel_dropout.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_batch_parallel_tensoradd.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_batchnorm_batch_parallel.py initial version 2020-03-27 22:54:54 +08:00
test_bias_add.py initial version 2020-03-27 22:54:54 +08:00
test_bn_prelu_cell.py initial version 2020-03-27 22:54:54 +08:00
test_bool_grad.py fix bug in infer_dtype function of hcom operations 2020-04-27 18:49:21 +08:00
test_broadcast_dict.py initial version 2020-03-27 22:54:54 +08:00
test_combined_tensor.py initial version 2020-03-27 22:54:54 +08:00
test_comparison_function_info.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_dataset.py initial version 2020-03-27 22:54:54 +08:00
test_dataset_interface.py fix sens shape error in loss_scale one_step_wrap 2020-05-12 14:09:51 +08:00
test_dataset_util.py initial version 2020-03-27 22:54:54 +08:00
test_dense_matmul.py initial version 2020-03-27 22:54:54 +08:00
test_different_type_for_div_op.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_dropout_do_mask.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_element_wise_function.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_expand_dims.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_forward_graph.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_gather_v2.py fix gatherv2 and dataset bug 2020-05-13 09:44:37 +08:00
test_gather_v2_primitive.py add gatherv2 distributed op 2020-05-09 17:20:50 +08:00
test_get_next.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_get_parameter_layout.py add slice shape for param info 2020-05-11 16:01:55 +08:00
test_hybird_parallel_activation.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_l2normalize.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_layer_norm.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_linear.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_load_tensor.py initial version 2020-03-27 22:54:54 +08:00
test_loop_two_matmul.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_loss_and_optimizer.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_matmul_dropout.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_matmul_tensor.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_mix_precision_hybrid_parallel.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_neg.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_one_dev.py remove ge depend in cpu 2020-04-02 19:38:42 +08:00
test_one_hot_net.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_one_weight_parameter.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_onehot.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_operator_model_parallel.py initial version 2020-03-27 22:54:54 +08:00
test_optimizer.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_optimizer_clone_weight.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_parameter_init.py initial version 2020-03-27 22:54:54 +08:00
test_prelu.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_prelu_cell.py initial version 2020-03-27 22:54:54 +08:00
test_reduce_method_info.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_reshape.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_reshape_parameter.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_scalar_loss.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_semi_auto_two_subgraphs.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_set_auto_parallel_context.py initial version 2020-03-27 22:54:54 +08:00
test_sigmoid_cross_entropy_with_logits.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_softmax_cross_entropy_expand.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_softmax_cross_entropy_loss.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_split_grad_sens.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_square.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_squeeze_info.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_step_parallel.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_strategy_checkpoint.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_sum_as_loss.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_transpose.py initial version 2020-03-27 22:54:54 +08:00
test_two_matmul.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_two_weights_parameter.py add parallel mode for cell 2020-05-07 20:11:20 +08:00
test_using_seed_for_initializer.py Support weight compile according to shape 2020-05-13 17:49:07 +08:00
test_virtual_dataset_3_input.py add a new vritualdataset cell for three inputs 2020-05-09 11:10:33 +08:00