You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
mindspore/tests/ut/python/parallel
jinyaohui 2907cf4488
remove some context param
5 years ago
..
parallel_end_to_end add parallel mode for cell 5 years ago
__init__.py implementing-searching-strategy-for-inference 5 years ago
conftest.py reset auto parallel context before each test 5 years ago
test_add_relu_redistribution.py add parallel mode for cell 5 years ago
test_allreduce_fusion.py enable/disable allreduce_fusion 5 years ago
test_alltoall.py remove ge depend in cpu 5 years ago
test_arithmetic.py add parallel mode for cell 5 years ago
test_auto_parallel_BN_PReLU.py add parallel mode for cell 5 years ago
test_auto_parallel_arithmetic.py add parallel mode for cell 5 years ago
test_auto_parallel_assign_sub_with_ref_key.py add parallel mode for cell 5 years ago
test_auto_parallel_cast.py add parallel mode for cell 5 years ago
test_auto_parallel_common_parameter.py add parallel mode for cell 5 years ago
test_auto_parallel_double_star.py add parallel mode for cell 5 years ago
test_auto_parallel_double_subgraphs.py add parallel mode for cell 5 years ago
test_auto_parallel_fc_nobias.py add parallel mode for cell 5 years ago
test_auto_parallel_four_matmul.py add parallel mode for cell 5 years ago
test_auto_parallel_inference.py calculating PEAK memory cost in the inference phase 5 years ago
test_auto_parallel_l2normalize.py add parallel mode for cell 5 years ago
test_auto_parallel_matmul_drop.py add parallel mode for cell 5 years ago
test_auto_parallel_matmul_prelu.py add parallel mode for cell 5 years ago
test_auto_parallel_onehot.py add parallel mode for cell 5 years ago
test_auto_parallel_parameter_cast.py add parallel mode for cell 5 years ago
test_auto_parallel_partial_strategy.py add parallel mode for cell 5 years ago
test_auto_parallel_reduce_method.py add parallel mode for cell 5 years ago
test_auto_parallel_reshape.py move InferStraByTensorInfo to tensor_info.h 5 years ago
test_auto_parallel_resnet.py remove some context param 5 years ago
test_auto_parallel_rhombus.py add parallel mode for cell 5 years ago
test_auto_parallel_softmax_loss.py add parallel mode for cell 5 years ago
test_auto_parallel_transformer.py add parallel mode for cell 5 years ago
test_auto_parallel_transpose.py add parallel mode for cell 5 years ago
test_auto_parallel_tuple_depend.py add parallel mode for cell 5 years ago
test_auto_parallel_two_bn.py add parallel mode for cell 5 years ago
test_auto_parallel_two_matmul.py add parallel mode for cell 5 years ago
test_auto_parallel_two_partial_matmul.py add parallel mode for cell 5 years ago
test_auto_parallel_zig_zag.py add parallel mode for cell 5 years ago
test_auto_star_elimination.py add parallel mode for cell 5 years ago
test_batch_matmul.py add parallel mode for cell 5 years ago
test_batch_parallel.py add parallel mode for cell 5 years ago
test_batch_parallel_dropout.py add parallel mode for cell 5 years ago
test_batch_parallel_tensoradd.py add parallel mode for cell 5 years ago
test_batchnorm_batch_parallel.py initial version 5 years ago
test_bias_add.py initial version 5 years ago
test_bn_prelu_cell.py initial version 5 years ago
test_bool_grad.py fix bug in infer_dtype function of hcom operations 5 years ago
test_broadcast_dict.py initial version 5 years ago
test_combined_tensor.py initial version 5 years ago
test_comparison_function_info.py add parallel mode for cell 5 years ago
test_dataset.py initial version 5 years ago
test_dataset_interface.py fix sens shape error in loss_scale one_step_wrap 5 years ago
test_dataset_util.py initial version 5 years ago
test_dense_matmul.py initial version 5 years ago
test_different_type_for_div_op.py add parallel mode for cell 5 years ago
test_dropout_do_mask.py add parallel mode for cell 5 years ago
test_element_wise_function.py add parallel mode for cell 5 years ago
test_expand_dims.py add parallel mode for cell 5 years ago
test_forward_graph.py add parallel mode for cell 5 years ago
test_gather_v2.py fix gatherv2 and dataset bug 5 years ago
test_gather_v2_primitive.py add gatherv2 distributed op 5 years ago
test_get_next.py add parallel mode for cell 5 years ago
test_get_parameter_layout.py add slice shape for param info 5 years ago
test_hybird_parallel_activation.py add parallel mode for cell 5 years ago
test_l2normalize.py add parallel mode for cell 5 years ago
test_layer_norm.py add parallel mode for cell 5 years ago
test_linear.py add parallel mode for cell 5 years ago
test_load_tensor.py initial version 5 years ago
test_loop_two_matmul.py add parallel mode for cell 5 years ago
test_loss_and_optimizer.py add parallel mode for cell 5 years ago
test_matmul_dropout.py add parallel mode for cell 5 years ago
test_matmul_tensor.py add parallel mode for cell 5 years ago
test_mix_precision_hybrid_parallel.py add parallel mode for cell 5 years ago
test_neg.py add parallel mode for cell 5 years ago
test_one_dev.py remove ge depend in cpu 5 years ago
test_one_hot_net.py add parallel mode for cell 5 years ago
test_one_weight_parameter.py add parallel mode for cell 5 years ago
test_onehot.py add parallel mode for cell 5 years ago
test_operator_model_parallel.py initial version 5 years ago
test_optimizer.py add parallel mode for cell 5 years ago
test_optimizer_clone_weight.py add parallel mode for cell 5 years ago
test_parameter_init.py initial version 5 years ago
test_prelu.py add parallel mode for cell 5 years ago
test_prelu_cell.py initial version 5 years ago
test_reduce_method_info.py add parallel mode for cell 5 years ago
test_reshape.py add parallel mode for cell 5 years ago
test_reshape_parameter.py add parallel mode for cell 5 years ago
test_scalar_loss.py add parallel mode for cell 5 years ago
test_semi_auto_two_subgraphs.py add parallel mode for cell 5 years ago
test_set_auto_parallel_context.py initial version 5 years ago
test_sigmoid_cross_entropy_with_logits.py add parallel mode for cell 5 years ago
test_softmax_cross_entropy_expand.py add parallel mode for cell 5 years ago
test_softmax_cross_entropy_loss.py add parallel mode for cell 5 years ago
test_split_grad_sens.py add parallel mode for cell 5 years ago
test_square.py add parallel mode for cell 5 years ago
test_squeeze_info.py add parallel mode for cell 5 years ago
test_step_parallel.py add parallel mode for cell 5 years ago
test_strategy_checkpoint.py add parallel mode for cell 5 years ago
test_sum_as_loss.py add parallel mode for cell 5 years ago
test_transpose.py initial version 5 years ago
test_two_matmul.py add parallel mode for cell 5 years ago
test_two_weights_parameter.py add parallel mode for cell 5 years ago
test_using_seed_for_initializer.py Support weight compile according to shape 5 years ago
test_virtual_dataset_3_input.py add a new vritualdataset cell for three inputs 5 years ago