• Home
Name Date Size #Lines LOC

..--

utils/06-May-2025-

__init__.pyD06-May-20251.5 KiB

conftest.pyD06-May-20251.2 KiB

test_adafactor.pyD06-May-20253.6 KiB

test_add_relu_redistribution.pyD06-May-20252.7 KiB

test_addn.pyD06-May-20252.4 KiB

test_allreduce_fusion.pyD06-May-202516 KiB

test_alltoall.pyD06-May-20259.4 KiB

test_anti_quant.pyD06-May-20253.7 KiB

test_arithmetic.pyD06-May-202546.9 KiB

test_asd.pyD06-May-20257.7 KiB

test_auto_parallel_BN_PReLU.pyD06-May-20252.4 KiB

test_auto_parallel_activation.pyD06-May-20252.1 KiB

test_auto_parallel_adasum.pyD06-May-20255.5 KiB

test_auto_parallel_arithmetic.pyD06-May-20256.9 KiB

test_auto_parallel_assign_sub_with_ref_key.pyD06-May-20252.4 KiB

test_auto_parallel_batch_matmul.pyD06-May-20252.9 KiB

test_auto_parallel_cast.pyD06-May-20253.3 KiB

test_auto_parallel_common_parameter.pyD06-May-20252.5 KiB

test_auto_parallel_data_parallel_mode.pyD06-May-20254.1 KiB

test_auto_parallel_double_sources.pyD06-May-20254.2 KiB

test_auto_parallel_double_star.pyD06-May-20253 KiB

test_auto_parallel_double_subgraphs.pyD06-May-20256.7 KiB

test_auto_parallel_fc_nobias.pyD06-May-20252.4 KiB

test_auto_parallel_flag.pyD06-May-20254.2 KiB

test_auto_parallel_for_loop.pyD06-May-20254.6 KiB

test_auto_parallel_for_loop_multi_subgraph.pyD06-May-20255.2 KiB

test_auto_parallel_for_loop_reshape.pyD06-May-20254.8 KiB

test_auto_parallel_for_loop_simplify.pyD06-May-20253.3 KiB

test_auto_parallel_four_matmul.pyD06-May-20254.5 KiB

test_auto_parallel_frontend_scheduling_with_pangu_alpha.pyD06-May-202531 KiB

test_auto_parallel_inference.pyD06-May-20252.1 KiB

test_auto_parallel_l2normalize.pyD06-May-20252.6 KiB

test_auto_parallel_matmul_drop.pyD06-May-20252.5 KiB

test_auto_parallel_matmul_prelu.pyD06-May-20252.9 KiB

test_auto_parallel_matmul_seq_parallel.pyD06-May-20255.9 KiB

test_auto_parallel_mem_coef_interface.pyD06-May-20251.6 KiB

test_auto_parallel_multi_graph.pyD06-May-20252.6 KiB

test_auto_parallel_onehot.pyD06-May-20254.6 KiB

test_auto_parallel_pangu_alpha.pyD06-May-202533.3 KiB

test_auto_parallel_parameter_cast.pyD06-May-20252.7 KiB

test_auto_parallel_partial_strategy.pyD06-May-20252.8 KiB

test_auto_parallel_rec_batch_gather.pyD06-May-20254 KiB

test_auto_parallel_rec_reducemin.pyD06-May-20252.4 KiB

test_auto_parallel_rec_reshape.pyD06-May-20255.3 KiB

test_auto_parallel_rec_softmax_loss.pyD06-May-20255.3 KiB

test_auto_parallel_rec_unknownop.pyD06-May-20256.7 KiB

test_auto_parallel_reduce_method.pyD06-May-20254.4 KiB

test_auto_parallel_reshape.pyD06-May-202513.6 KiB

test_auto_parallel_resnet_sharding_propagation.pyD06-May-202511.5 KiB

test_auto_parallel_resnet_sharding_propagation2.pyD06-May-202511.1 KiB

test_auto_parallel_rhombus.pyD06-May-20255.3 KiB

test_auto_parallel_sapp_RmsNorm.pyD06-May-20252.3 KiB

test_auto_parallel_sapp_batchparallel.pyD06-May-20251.8 KiB

test_auto_parallel_sapp_broadcast.pyD06-May-20251.8 KiB

test_auto_parallel_sapp_flash_attention.pyD06-May-20257.4 KiB

test_auto_parallel_sapp_gather.pyD06-May-20254.1 KiB

test_auto_parallel_sapp_opt_restrict.pyD06-May-20251.9 KiB

test_auto_parallel_sapp_strided_slice.pyD06-May-20252.8 KiB

test_auto_parallel_segment_min.pyD06-May-20252.4 KiB

test_auto_parallel_segment_sum.pyD06-May-20252.4 KiB

test_auto_parallel_shard_propagation.pyD06-May-20253.1 KiB

test_auto_parallel_shard_propagation2.pyD06-May-20252.2 KiB

test_auto_parallel_shard_propagation3.pyD06-May-20252.6 KiB

test_auto_parallel_share_param.pyD06-May-20254.4 KiB

test_auto_parallel_softmax_loss.pyD06-May-20252.4 KiB

test_auto_parallel_star_partial_strategy.pyD06-May-20255.7 KiB

test_auto_parallel_transpose.pyD06-May-20253.1 KiB

test_auto_parallel_triangle_overwrite.pyD06-May-20252.8 KiB

test_auto_parallel_tril.pyD06-May-20252.3 KiB

test_auto_parallel_tuple_depend.pyD06-May-20252.8 KiB

test_auto_parallel_two_bn.pyD06-May-20253 KiB

test_auto_parallel_two_matmul.pyD06-May-20257.6 KiB

test_auto_parallel_two_partial_matmul.pyD06-May-20252.5 KiB

test_auto_parallel_zig_zag.pyD06-May-20252.7 KiB

test_auto_pipeline.pyD06-May-20251.4 KiB

test_auto_star_elimination.pyD06-May-20253.2 KiB

test_autoparallel_rec_shared_param_strmodif.pyD06-May-20253.9 KiB

test_avgpool_3d.pyD06-May-20257.9 KiB

test_batch_matmul.pyD06-May-20254.9 KiB

test_batch_matmul_high_dimension.pyD06-May-202510.5 KiB

test_batch_parallel.pyD06-May-20254.4 KiB

test_batch_parallel_dropout.pyD06-May-20252.3 KiB

test_batch_parallel_tensoradd.pyD06-May-20252.2 KiB

test_batchmm.pyD06-May-20253.5 KiB

test_batchnorm.pyD06-May-20257.2 KiB

test_batchnorm_batch_parallel.pyD06-May-20254.8 KiB

test_batchparallel_replace_shape.pyD06-May-20252.9 KiB

test_bias_add.pyD06-May-20253.1 KiB

test_bias_add_comm_swap.pyD06-May-20254.1 KiB

test_bool_grad.pyD06-May-20252.4 KiB

test_bounding_box_encode.pyD06-May-20252.7 KiB

test_broadcast_dict.pyD06-May-20252.8 KiB

test_broadcast_to.pyD06-May-20254.2 KiB

test_cdist.pyD06-May-20254.7 KiB

test_cell_shard_check.pyD06-May-20257.2 KiB

test_checkvalid.pyD06-May-20254.2 KiB

test_combined_tensor.pyD06-May-20252.2 KiB

test_comm_fusion.pyD06-May-202510.5 KiB

test_comm_not_recompute.pyD06-May-20254.4 KiB

test_comparison_function_info.pyD06-May-202513.7 KiB

test_concat.pyD06-May-20256 KiB

test_config_layout.pyD06-May-20254.6 KiB

test_config_layout_for_arbitrary_ops.pyD06-May-20258.5 KiB

test_conformer.pyD06-May-202535.7 KiB

test_control_flow.pyD06-May-20253.2 KiB

test_conv2d.pyD06-May-202521.7 KiB

test_conv2d_transpose.pyD06-May-202513.5 KiB

test_conv3d.pyD06-May-20256.6 KiB

test_convolution.pyD06-May-20254.3 KiB

test_crop_and_resize.pyD06-May-20253.6 KiB

test_cum_ops.pyD06-May-20253.3 KiB

test_cus_matmul_dds.pyD06-May-20257.1 KiB

test_dataset.pyD06-May-20251.5 KiB

test_dataset_interface.pyD06-May-20255.8 KiB

test_decoder_k_v_cache.pyD06-May-20256.7 KiB

test_dense_matmul.pyD06-May-20252 KiB

test_dequant.pyD06-May-20251.9 KiB

test_device_num_not_power_of_2.pyD06-May-20257.5 KiB

test_different_type_for_div_op.pyD06-May-20253.6 KiB

test_distribute_predict.pyD06-May-20253.9 KiB

test_dropout.pyD06-May-20253.3 KiB

test_dropout_do_mask.pyD06-May-20254.4 KiB

test_dsd_matmul.pyD06-May-20257.2 KiB

test_dynamic_base_operation.pyD06-May-202521.8 KiB

test_dynamic_broadcast_to.pyD06-May-20252.9 KiB

test_dynamic_bs_data_parallel.pyD06-May-20253.3 KiB

test_dynamic_data_sink.pyD06-May-20258.7 KiB

test_dynamic_matmul.pyD06-May-20252 KiB

test_dynamic_pangu_alpha.pyD06-May-202571.4 KiB

test_dynamic_pangu_alpha_without_skip_redistribution.pyD06-May-202570.7 KiB

test_dynamic_quant.pyD06-May-20253.1 KiB

test_dynamic_shape.pyD06-May-20254.8 KiB

test_dynamic_shape_graph_utils.pyD06-May-202564.8 KiB

test_dynamic_shape_merge_allgather.pyD06-May-20254.8 KiB

test_dynamic_shape_multi_axis.pyD06-May-202525.1 KiB

test_dynamic_shape_no_redistribution.pyD06-May-202525.4 KiB

test_dynamic_stridedslice.pyD06-May-20253.3 KiB

test_element_wise_axis.pyD06-May-20252.1 KiB

test_element_wise_function.pyD06-May-202545 KiB

test_element_wise_three_inputs.pyD06-May-20252.2 KiB

test_element_wise_two_inputs.pyD06-May-20253.5 KiB

test_embeddinglookup.pyD06-May-20253.7 KiB

test_eval.pyD06-May-20253.1 KiB

test_expand_dims.pyD06-May-20254.1 KiB

test_ffn.pyD06-May-202511.3 KiB

test_ffn_quant.pyD06-May-20256.9 KiB

test_ffnext.pyD06-May-202511.3 KiB

test_ffnext_quant.pyD06-May-20256.9 KiB

test_fillv2.pyD06-May-20251.9 KiB

test_finegraint_micro_interleaved.pyD06-May-202520.2 KiB

test_flash_attention_score_info.pyD06-May-202521.6 KiB

test_fold_pipeline_shared_parameter.pyD06-May-20258.9 KiB

test_fold_pipeline_split.pyD06-May-202515.6 KiB

test_forward_graph.pyD06-May-20253.1 KiB

test_full_batch.pyD06-May-20253.9 KiB

test_func_mode.pyD06-May-20254.4 KiB

test_func_mode_pipeline.pyD06-May-20255.5 KiB

test_fused_infer_attention_score.pyD06-May-202518.5 KiB

test_gamma.pyD06-May-20252.9 KiB

test_gather_batch_dims.pyD06-May-20257.6 KiB

test_gather_field_split.pyD06-May-20254.8 KiB

test_gather_v2.pyD06-May-202519 KiB

test_gatherd.pyD06-May-20254.5 KiB

test_gathernd.pyD06-May-20253.5 KiB

test_gathernd_further.pyD06-May-202513 KiB

test_generate_batch_parallel.pyD06-May-20258.7 KiB

test_get_next.pyD06-May-20255.2 KiB

test_get_parameter_layout.pyD06-May-20252.6 KiB

test_global_norm.pyD06-May-20259.5 KiB

test_gpu_dropout.pyD06-May-20254 KiB

test_grad_accumulation.pyD06-May-20258.2 KiB

test_grid_sampler2d.pyD06-May-20254.6 KiB

test_group_norm.pyD06-May-20254 KiB

test_grouped_matmul.pyD06-May-202514 KiB

test_grouped_pairwise_exchange_alltoall.pyD06-May-20254.3 KiB

test_hybird_parallel_activation.pyD06-May-202510.2 KiB

test_incre_flash_attention.pyD06-May-20258.4 KiB

test_index_select.pyD06-May-20254.2 KiB

test_initializer_weight_slice.pyD06-May-20256.3 KiB

test_inplace_op.pyD06-May-20253.8 KiB

test_invert_populationcount.pyD06-May-20252.3 KiB

test_iou.pyD06-May-20252.5 KiB

test_kldiv_loss.pyD06-May-20254.9 KiB

test_kv_cache_scatter_update.pyD06-May-20253 KiB

test_l2_loss.pyD06-May-20252.5 KiB

test_l2normalize.pyD06-May-20252.6 KiB

test_layer_norm.pyD06-May-20254.5 KiB

test_layer_norm_further.pyD06-May-202515.7 KiB

test_layout_compare.pyD06-May-20253.6 KiB

test_layout_extend.pyD06-May-202515 KiB

test_layout_extend_activation.pyD06-May-20253.9 KiB

test_layout_extend_arithmetic.pyD06-May-20256.9 KiB

test_layout_extend_bias_add.pyD06-May-20254.1 KiB

test_layout_extend_layernorm.pyD06-May-20255.7 KiB

test_layout_extend_rmsnorm.pyD06-May-20255.1 KiB

test_layout_extend_transformer.pyD06-May-202513.7 KiB

test_layout_extend_transpose.pyD06-May-20255 KiB

test_lerp.pyD06-May-20256 KiB

test_lin_space.pyD06-May-20254.2 KiB

test_lin_space_ext.pyD06-May-20256.2 KiB

test_linear.pyD06-May-20252.6 KiB

test_load_tensor.pyD06-May-20251.4 KiB

test_loop_two_matmul.pyD06-May-20253 KiB

test_loss_and_o2_level.pyD06-May-20254.3 KiB

test_loss_and_optimizer.pyD06-May-20257.2 KiB

test_loss_scale.pyD06-May-20257.1 KiB

test_manual_embedding_lookup.pyD06-May-20257.7 KiB

test_manual_gatherv2.pyD06-May-20259.7 KiB

test_matmul_add_comm_reduction.pyD06-May-20259 KiB

test_matmul_dropout.pyD06-May-20254.5 KiB

test_matmul_tensor.pyD06-May-20255.3 KiB

test_matmul_transpose_attr.pyD06-May-202516 KiB

test_maxpool_3d.pyD06-May-20257.8 KiB

test_maxpool_avgpool.pyD06-May-20258.8 KiB

test_micro_batch_interleaved.pyD06-May-20252.5 KiB

test_mirror_insert.pyD06-May-20255.3 KiB

test_mix_precision_hybrid_parallel.pyD06-May-20253 KiB

test_model_flops_cal.pyD06-May-202515.2 KiB

test_model_with_loss.pyD06-May-20253.9 KiB

test_model_without_loss.pyD06-May-20256.5 KiB

test_moe_finalize_routing.pyD06-May-20253.2 KiB

test_mul_div_bn.pyD06-May-20252.9 KiB

test_mul_softmax_net.pyD06-May-20258.3 KiB

test_multi_field_embedding.pyD06-May-20255.5 KiB

test_neg.pyD06-May-20253.1 KiB

test_neighborexchange.pyD06-May-202518.2 KiB

test_neighborexchangev2.pyD06-May-202531.4 KiB

test_o2_level.pyD06-May-20257.1 KiB

test_one_dev.pyD06-May-20254.4 KiB

test_one_hot_net.pyD06-May-202513.7 KiB

test_one_weight_parameter.pyD06-May-20252.6 KiB

test_onehot.pyD06-May-20255.7 KiB

test_onehot_2dim.pyD06-May-20252.4 KiB

test_operator_model_parallel.pyD06-May-202512.8 KiB

test_optimizer.pyD06-May-20252.6 KiB

test_optimizer_clone_weight.pyD06-May-20253.6 KiB

test_pack.pyD06-May-202512.7 KiB

test_pad_v3.pyD06-May-20256.5 KiB

test_paged_attention.pyD06-May-20255.1 KiB

test_paged_attention_mask.pyD06-May-20255.8 KiB

test_parallel_mirror_group.pyD06-May-20259.4 KiB

test_parallel_optimizer.pyD06-May-202516.9 KiB

test_parallel_optimizer_shared_cast.pyD06-May-20256.7 KiB

test_parallel_optimizer_without_grad.pyD06-May-20254.9 KiB

test_parallel_while.pyD06-May-20253.5 KiB

test_param_broadcast_cell.pyD06-May-20251.6 KiB

test_parameter_clone.pyD06-May-20251.3 KiB

test_parameter_init.pyD06-May-20251.9 KiB

test_parameter_merge.pyD06-May-20253.5 KiB

test_parameter_multi_users.pyD06-May-20253.6 KiB

test_pipeline_dynamic_shape.pyD06-May-20255.6 KiB

test_pipeline_end_node.pyD06-May-20253.2 KiB

test_pipeline_grad_freeze.pyD06-May-20254.7 KiB

test_pipeline_grad_reducer.pyD06-May-20255.2 KiB

test_pipeline_inference.pyD06-May-20258.5 KiB

test_pipeline_interleave.pyD06-May-20256.9 KiB

test_pipeline_interleave_multi_send_recv.pyD06-May-20254 KiB

test_pipeline_multiout.pyD06-May-20253.9 KiB

test_pipeline_opt_detection.pyD06-May-202514.7 KiB

test_pipeline_remove_node.pyD06-May-20254.8 KiB

test_pipeline_shard_with_mirror.pyD06-May-20254.4 KiB

test_pipeline_shared_parameter.pyD06-May-202516.9 KiB

test_pipeline_split.pyD06-May-202533 KiB

test_prelu.pyD06-May-20255.4 KiB

test_prelu_cell.pyD06-May-20252.8 KiB

test_print.pyD06-May-20256.1 KiB

test_prompt_flash_attention.pyD06-May-202514.4 KiB

test_prompt_k_v_cache.pyD06-May-20255.7 KiB

test_quant.pyD06-May-20253.4 KiB

test_quant_batch_matmul.pyD06-May-20258.8 KiB

test_quant_ops.pyD06-May-20256.4 KiB

test_quant_v2.pyD06-May-20254.4 KiB

test_random_choice_with_mask.pyD06-May-20252.3 KiB

test_range.pyD06-May-20253.5 KiB

test_range_dynamic.pyD06-May-20252.2 KiB

test_reduce_method_info.pyD06-May-202537.9 KiB

test_remove_cast_before_assign_add.pyD06-May-20251.9 KiB

test_repeated_calc.pyD06-May-20253.4 KiB

test_reshape.pyD06-May-202522.7 KiB

test_reshape_and_cache.pyD06-May-20255 KiB

test_reshape_consecutive.pyD06-May-20252.4 KiB

test_reshape_optimized.pyD06-May-20252 KiB

test_reshape_parameter.pyD06-May-20252.7 KiB

test_reshape_parameter_scalar_or_other_shape.pyD06-May-20256.6 KiB

test_reshape_shard_propagation.pyD06-May-202513.4 KiB

test_reshape_skip_redistribution.pyD06-May-20252.2 KiB

test_reshape_unexpand.pyD06-May-202510.8 KiB

test_resizebilinear.pyD06-May-202510.9 KiB

test_ring_attention.pyD06-May-20256.4 KiB

test_rms_norm.pyD06-May-20256.4 KiB

test_roi_align.pyD06-May-20255.1 KiB

test_scalar_loss.pyD06-May-20252.1 KiB

test_scatter_nd_ops.pyD06-May-202512.9 KiB

test_scatter_ops.pyD06-May-202510 KiB

test_scatter_update.pyD06-May-20253.3 KiB

test_select.pyD06-May-20253.9 KiB

test_self_attention.pyD06-May-20254.7 KiB

test_semi_auto_two_subgraphs.pyD06-May-20254.2 KiB

test_set_auto_parallel_context.pyD06-May-20256.1 KiB

test_set_offload_context.pyD06-May-20252 KiB

test_set_repeated_num_in_dev_matrix_right.pyD06-May-20252.5 KiB

test_set_stridedslice_strategy.pyD06-May-20252.6 KiB

test_sharding_propagation_for_conv2ds.pyD06-May-20256.6 KiB

test_shared_param_and_mix_precision.pyD06-May-20254.3 KiB

test_sigmoid_cross_entropy_with_logits.pyD06-May-20253.3 KiB

test_slice.pyD06-May-20256.3 KiB

test_softmax_cross_entropy_loss.pyD06-May-20254.1 KiB

test_softmax_gather_net.pyD06-May-202516 KiB

test_sparse_feature_bprop.pyD06-May-20254.5 KiB

test_sparse_gather_v2.pyD06-May-20257.2 KiB

test_split.pyD06-May-20255.4 KiB

test_split_ext.pyD06-May-20255.3 KiB

test_split_grad_sens.pyD06-May-20256.7 KiB

test_splitv.pyD06-May-20254.2 KiB

test_square.pyD06-May-20253.2 KiB

test_squeeze_info.pyD06-May-20252.8 KiB

test_stand_alone.pyD06-May-20253.5 KiB

test_step_parallel.pyD06-May-20252.7 KiB

test_strategy_checkpoint.pyD06-May-202513.7 KiB

test_stridedslice.pyD06-May-202527.1 KiB

test_sum_as_loss.pyD06-May-20252.8 KiB

test_syncbatchnorm.pyD06-May-20252.8 KiB

test_tile.pyD06-May-20255.8 KiB

test_topk.pyD06-May-20253.3 KiB

test_topk_ext.pyD06-May-20253.8 KiB

test_topkrouter.pyD06-May-20252.3 KiB

test_train_and_eval.pyD06-May-20252.5 KiB

test_transform_tensor.pyD06-May-202525.1 KiB

test_transpose.pyD06-May-20253.6 KiB

test_two_matmul.pyD06-May-202521.5 KiB

test_two_weights_parameter.pyD06-May-20252.9 KiB

test_uniform_candidate_sampler.pyD06-May-20257.1 KiB

test_uniform_real.pyD06-May-20252.5 KiB

test_unsortedsegmentmax.pyD06-May-20255.5 KiB

test_unsortedsegmentmin.pyD06-May-20255.5 KiB

test_unsortedsegmentprod.pyD06-May-20256.8 KiB

test_unsortedsegmentsum.pyD06-May-20258.1 KiB

test_using_seed_for_initializer.pyD06-May-20252.8 KiB

test_virtual_dataset_3_input.pyD06-May-20252.7 KiB

test_virtual_dataset_with_strategy.pyD06-May-202513.5 KiB

test_virtual_output.pyD06-May-202514.9 KiB

test_weight_quant_batch_matmul.pyD06-May-202514.5 KiB

test_wrong_parallel_mode_for_grad_reducer.pyD06-May-20254.8 KiB