• Home
Name Date Size #Lines LOC

..--

utils/22-Oct-2025-

__init__.pyD22-Oct-20251.5 KiB

conftest.pyD22-Oct-20251.2 KiB

test_adafactor.pyD22-Oct-20253.6 KiB

test_add_relu_redistribution.pyD22-Oct-20252.7 KiB

test_addn.pyD22-Oct-20252.4 KiB

test_allreduce_fusion.pyD22-Oct-202516 KiB

test_alltoall.pyD22-Oct-20259.4 KiB

test_anti_quant.pyD22-Oct-20253.7 KiB

test_arithmetic.pyD22-Oct-202546.9 KiB

test_asd.pyD22-Oct-20257.7 KiB

test_auto_parallel_BN_PReLU.pyD22-Oct-20252.4 KiB

test_auto_parallel_activation.pyD22-Oct-20252.1 KiB

test_auto_parallel_adasum.pyD22-Oct-20255.5 KiB

test_auto_parallel_arithmetic.pyD22-Oct-20256.9 KiB

test_auto_parallel_assign_sub_with_ref_key.pyD22-Oct-20252.4 KiB

test_auto_parallel_batch_matmul.pyD22-Oct-20252.9 KiB

test_auto_parallel_cast.pyD22-Oct-20253.3 KiB

test_auto_parallel_common_parameter.pyD22-Oct-20252.5 KiB

test_auto_parallel_data_parallel_mode.pyD22-Oct-20254.1 KiB

test_auto_parallel_double_sources.pyD22-Oct-20254.2 KiB

test_auto_parallel_double_star.pyD22-Oct-20253 KiB

test_auto_parallel_double_subgraphs.pyD22-Oct-20256.7 KiB

test_auto_parallel_fc_nobias.pyD22-Oct-20252.4 KiB

test_auto_parallel_flag.pyD22-Oct-20254.2 KiB

test_auto_parallel_for_loop.pyD22-Oct-20254.6 KiB

test_auto_parallel_for_loop_multi_subgraph.pyD22-Oct-20255.2 KiB

test_auto_parallel_for_loop_reshape.pyD22-Oct-20254.8 KiB

test_auto_parallel_for_loop_simplify.pyD22-Oct-20253.3 KiB

test_auto_parallel_four_matmul.pyD22-Oct-20254.5 KiB

test_auto_parallel_frontend_scheduling_with_pangu_alpha.pyD22-Oct-202531 KiB

test_auto_parallel_inference.pyD22-Oct-20252.1 KiB

test_auto_parallel_l2normalize.pyD22-Oct-20252.6 KiB

test_auto_parallel_matmul_drop.pyD22-Oct-20252.5 KiB

test_auto_parallel_matmul_prelu.pyD22-Oct-20252.9 KiB

test_auto_parallel_matmul_seq_parallel.pyD22-Oct-20255.9 KiB

test_auto_parallel_mem_coef_interface.pyD22-Oct-20251.6 KiB

test_auto_parallel_multi_graph.pyD22-Oct-20252.6 KiB

test_auto_parallel_onehot.pyD22-Oct-20254.6 KiB

test_auto_parallel_pangu_alpha.pyD22-Oct-202533.3 KiB

test_auto_parallel_parameter_cast.pyD22-Oct-20252.7 KiB

test_auto_parallel_partial_strategy.pyD22-Oct-20252.8 KiB

test_auto_parallel_rec_batch_gather.pyD22-Oct-20254 KiB

test_auto_parallel_rec_reducemin.pyD22-Oct-20252.4 KiB

test_auto_parallel_rec_reshape.pyD22-Oct-20255.3 KiB

test_auto_parallel_rec_softmax_loss.pyD22-Oct-20255.3 KiB

test_auto_parallel_rec_unknownop.pyD22-Oct-20256.7 KiB

test_auto_parallel_reduce_method.pyD22-Oct-20254.4 KiB

test_auto_parallel_reshape.pyD22-Oct-202513.6 KiB

test_auto_parallel_resnet_sharding_propagation.pyD22-Oct-202511.5 KiB

test_auto_parallel_resnet_sharding_propagation2.pyD22-Oct-202511.1 KiB

test_auto_parallel_rhombus.pyD22-Oct-20255.3 KiB

test_auto_parallel_sapp_RmsNorm.pyD22-Oct-20252.3 KiB

test_auto_parallel_sapp_batchparallel.pyD22-Oct-20251.8 KiB

test_auto_parallel_sapp_broadcast.pyD22-Oct-20251.8 KiB

test_auto_parallel_sapp_flash_attention.pyD22-Oct-20257.4 KiB

test_auto_parallel_sapp_gather.pyD22-Oct-20254.1 KiB

test_auto_parallel_sapp_opt_restrict.pyD22-Oct-20251.9 KiB

test_auto_parallel_sapp_strided_slice.pyD22-Oct-20252.8 KiB

test_auto_parallel_segment_min.pyD22-Oct-20252.4 KiB

test_auto_parallel_segment_sum.pyD22-Oct-20252.4 KiB

test_auto_parallel_shard_propagation.pyD22-Oct-20253.1 KiB

test_auto_parallel_shard_propagation2.pyD22-Oct-20252.2 KiB

test_auto_parallel_shard_propagation3.pyD22-Oct-20252.6 KiB

test_auto_parallel_share_param.pyD22-Oct-20254.4 KiB

test_auto_parallel_softmax_loss.pyD22-Oct-20252.4 KiB

test_auto_parallel_star_partial_strategy.pyD22-Oct-20255.7 KiB

test_auto_parallel_transpose.pyD22-Oct-20253.1 KiB

test_auto_parallel_triangle_overwrite.pyD22-Oct-20252.8 KiB

test_auto_parallel_tril.pyD22-Oct-20252.3 KiB

test_auto_parallel_tuple_depend.pyD22-Oct-20252.8 KiB

test_auto_parallel_two_bn.pyD22-Oct-20253 KiB

test_auto_parallel_two_matmul.pyD22-Oct-20257.6 KiB

test_auto_parallel_two_partial_matmul.pyD22-Oct-20252.5 KiB

test_auto_parallel_zig_zag.pyD22-Oct-20252.7 KiB

test_auto_pipeline.pyD22-Oct-20251.4 KiB

test_auto_star_elimination.pyD22-Oct-20253.2 KiB

test_autoparallel_rec_shared_param_strmodif.pyD22-Oct-20253.9 KiB

test_avgpool_3d.pyD22-Oct-20257.9 KiB

test_batch_matmul.pyD22-Oct-20254.9 KiB

test_batch_matmul_high_dimension.pyD22-Oct-202510.5 KiB

test_batch_parallel.pyD22-Oct-20254.4 KiB

test_batch_parallel_dropout.pyD22-Oct-20252.3 KiB

test_batch_parallel_tensoradd.pyD22-Oct-20252.2 KiB

test_batchmm.pyD22-Oct-20253.5 KiB

test_batchnorm.pyD22-Oct-20257.2 KiB

test_batchnorm_batch_parallel.pyD22-Oct-20254.8 KiB

test_batchparallel_replace_shape.pyD22-Oct-20252.9 KiB

test_bias_add.pyD22-Oct-20253.1 KiB

test_bias_add_comm_swap.pyD22-Oct-20254.1 KiB

test_bool_grad.pyD22-Oct-20252.4 KiB

test_bounding_box_encode.pyD22-Oct-20252.7 KiB

test_broadcast_dict.pyD22-Oct-20252.8 KiB

test_broadcast_to.pyD22-Oct-20254.2 KiB

test_cdist.pyD22-Oct-20254.7 KiB

test_cell_shard_check.pyD22-Oct-20257.2 KiB

test_checkvalid.pyD22-Oct-20254.2 KiB

test_combined_tensor.pyD22-Oct-20252.2 KiB

test_comm_fusion.pyD22-Oct-202510.5 KiB

test_comm_not_recompute.pyD22-Oct-20254.4 KiB

test_comparison_function_info.pyD22-Oct-202513.7 KiB

test_concat.pyD22-Oct-20256 KiB

test_config_layout.pyD22-Oct-20254.6 KiB

test_config_layout_for_arbitrary_ops.pyD22-Oct-20258.5 KiB

test_conformer.pyD22-Oct-202535.7 KiB

test_control_flow.pyD22-Oct-20253.2 KiB

test_conv2d.pyD22-Oct-202521.7 KiB

test_conv2d_transpose.pyD22-Oct-202513.5 KiB

test_conv3d.pyD22-Oct-20256.6 KiB

test_convolution.pyD22-Oct-20254.3 KiB

test_crop_and_resize.pyD22-Oct-20253.6 KiB

test_cum_ops.pyD22-Oct-20253.3 KiB

test_cus_matmul_dds.pyD22-Oct-20257.1 KiB

test_dataset.pyD22-Oct-20251.5 KiB

test_dataset_interface.pyD22-Oct-20255.8 KiB

test_decoder_k_v_cache.pyD22-Oct-20256.7 KiB

test_dense_matmul.pyD22-Oct-20252 KiB

test_dequant.pyD22-Oct-20251.9 KiB

test_device_num_not_power_of_2.pyD22-Oct-20257.5 KiB

test_different_type_for_div_op.pyD22-Oct-20253.6 KiB

test_distribute_predict.pyD22-Oct-20253.9 KiB

test_dropout.pyD22-Oct-20253.3 KiB

test_dropout_do_mask.pyD22-Oct-20254.4 KiB

test_dsd_matmul.pyD22-Oct-20257.2 KiB

test_dynamic_base_operation.pyD22-Oct-202521.8 KiB

test_dynamic_broadcast_to.pyD22-Oct-20252.9 KiB

test_dynamic_bs_data_parallel.pyD22-Oct-20253.3 KiB

test_dynamic_data_sink.pyD22-Oct-20258.7 KiB

test_dynamic_matmul.pyD22-Oct-20252 KiB

test_dynamic_pangu_alpha.pyD22-Oct-202571.4 KiB

test_dynamic_pangu_alpha_without_skip_redistribution.pyD22-Oct-202570.7 KiB

test_dynamic_quant.pyD22-Oct-20253.1 KiB

test_dynamic_shape.pyD22-Oct-20254.8 KiB

test_dynamic_shape_graph_utils.pyD22-Oct-202564.8 KiB

test_dynamic_shape_merge_allgather.pyD22-Oct-20254.8 KiB

test_dynamic_shape_multi_axis.pyD22-Oct-202525.1 KiB

test_dynamic_shape_no_redistribution.pyD22-Oct-202525.4 KiB

test_dynamic_stridedslice.pyD22-Oct-20253.3 KiB

test_element_wise_axis.pyD22-Oct-20252.1 KiB

test_element_wise_function.pyD22-Oct-202545 KiB

test_element_wise_three_inputs.pyD22-Oct-20252.2 KiB

test_element_wise_two_inputs.pyD22-Oct-20253.5 KiB

test_embeddinglookup.pyD22-Oct-20253.7 KiB

test_eval.pyD22-Oct-20253.1 KiB

test_expand_dims.pyD22-Oct-20254.1 KiB

test_ffn.pyD22-Oct-202511.3 KiB

test_ffn_quant.pyD22-Oct-20256.9 KiB

test_ffnext.pyD22-Oct-202511.3 KiB

test_ffnext_quant.pyD22-Oct-20256.9 KiB

test_fillv2.pyD22-Oct-20251.9 KiB

test_finegraint_micro_interleaved.pyD22-Oct-202520.2 KiB

test_flash_attention_score_info.pyD22-Oct-202521.6 KiB

test_fold_pipeline_shared_parameter.pyD22-Oct-20258.9 KiB

test_fold_pipeline_split.pyD22-Oct-202515.6 KiB

test_forward_graph.pyD22-Oct-20253.1 KiB

test_full_batch.pyD22-Oct-20253.9 KiB

test_func_mode.pyD22-Oct-20254.4 KiB

test_func_mode_pipeline.pyD22-Oct-20255.5 KiB

test_fused_infer_attention_score.pyD22-Oct-202518.5 KiB

test_gamma.pyD22-Oct-20252.9 KiB

test_gather_batch_dims.pyD22-Oct-20257.6 KiB

test_gather_field_split.pyD22-Oct-20254.8 KiB

test_gather_v2.pyD22-Oct-202519 KiB

test_gatherd.pyD22-Oct-20254.5 KiB

test_gathernd.pyD22-Oct-20253.5 KiB

test_gathernd_further.pyD22-Oct-202513 KiB

test_generate_batch_parallel.pyD22-Oct-20258.7 KiB

test_get_next.pyD22-Oct-20255.2 KiB

test_get_parameter_layout.pyD22-Oct-20252.6 KiB

test_global_norm.pyD22-Oct-20259.5 KiB

test_gpu_dropout.pyD22-Oct-20254 KiB

test_grad_accumulation.pyD22-Oct-20258.2 KiB

test_grid_sampler2d.pyD22-Oct-20254.6 KiB

test_group_norm.pyD22-Oct-20254 KiB

test_grouped_matmul.pyD22-Oct-202514 KiB

test_grouped_pairwise_exchange_alltoall.pyD22-Oct-20254.3 KiB

test_hybird_parallel_activation.pyD22-Oct-202510.2 KiB

test_incre_flash_attention.pyD22-Oct-20258.4 KiB

test_index_select.pyD22-Oct-20254.2 KiB

test_initializer_weight_slice.pyD22-Oct-20256.3 KiB

test_inplace_op.pyD22-Oct-20253.8 KiB

test_invert_populationcount.pyD22-Oct-20252.3 KiB

test_iou.pyD22-Oct-20252.5 KiB

test_kldiv_loss.pyD22-Oct-20254.9 KiB

test_kv_cache_scatter_update.pyD22-Oct-20253 KiB

test_l2_loss.pyD22-Oct-20252.5 KiB

test_l2normalize.pyD22-Oct-20252.6 KiB

test_layer_norm.pyD22-Oct-20254.5 KiB

test_layer_norm_further.pyD22-Oct-202515.7 KiB

test_layout_compare.pyD22-Oct-20253.6 KiB

test_layout_extend.pyD22-Oct-202515 KiB

test_layout_extend_activation.pyD22-Oct-20253.9 KiB

test_layout_extend_arithmetic.pyD22-Oct-20256.9 KiB

test_layout_extend_bias_add.pyD22-Oct-20254.1 KiB

test_layout_extend_layernorm.pyD22-Oct-20255.7 KiB

test_layout_extend_rmsnorm.pyD22-Oct-20255.1 KiB

test_layout_extend_transformer.pyD22-Oct-202513.7 KiB

test_layout_extend_transpose.pyD22-Oct-20255 KiB

test_lerp.pyD22-Oct-20256 KiB

test_lin_space.pyD22-Oct-20254.2 KiB

test_lin_space_ext.pyD22-Oct-20256.2 KiB

test_linear.pyD22-Oct-20252.6 KiB

test_load_tensor.pyD22-Oct-20251.4 KiB

test_loop_two_matmul.pyD22-Oct-20253 KiB

test_loss_and_o2_level.pyD22-Oct-20254.3 KiB

test_loss_and_optimizer.pyD22-Oct-20257.2 KiB

test_loss_scale.pyD22-Oct-20257.1 KiB

test_manual_embedding_lookup.pyD22-Oct-20257.7 KiB

test_manual_gatherv2.pyD22-Oct-20259.7 KiB

test_matmul_add_comm_reduction.pyD22-Oct-20259 KiB

test_matmul_dropout.pyD22-Oct-20254.5 KiB

test_matmul_tensor.pyD22-Oct-20255.3 KiB

test_matmul_transpose_attr.pyD22-Oct-202516 KiB

test_maxpool_3d.pyD22-Oct-20257.8 KiB

test_maxpool_avgpool.pyD22-Oct-20258.8 KiB

test_micro_batch_interleaved.pyD22-Oct-20252.5 KiB

test_mirror_insert.pyD22-Oct-20255.3 KiB

test_mix_precision_hybrid_parallel.pyD22-Oct-20253 KiB

test_model_flops_cal.pyD22-Oct-202515.2 KiB

test_model_with_loss.pyD22-Oct-20253.9 KiB

test_model_without_loss.pyD22-Oct-20256.5 KiB

test_moe_finalize_routing.pyD22-Oct-20253.2 KiB

test_mul_div_bn.pyD22-Oct-20252.9 KiB

test_mul_softmax_net.pyD22-Oct-20258.3 KiB

test_multi_field_embedding.pyD22-Oct-20255.5 KiB

test_neg.pyD22-Oct-20253.1 KiB

test_neighborexchange.pyD22-Oct-202518.2 KiB

test_neighborexchangev2.pyD22-Oct-202531.4 KiB

test_o2_level.pyD22-Oct-20257.1 KiB

test_one_dev.pyD22-Oct-20254.4 KiB

test_one_hot_net.pyD22-Oct-202513.7 KiB

test_one_weight_parameter.pyD22-Oct-20252.6 KiB

test_onehot.pyD22-Oct-20255.7 KiB

test_onehot_2dim.pyD22-Oct-20252.4 KiB

test_operator_model_parallel.pyD22-Oct-202512.8 KiB

test_optimizer.pyD22-Oct-20252.6 KiB

test_optimizer_clone_weight.pyD22-Oct-20253.6 KiB

test_pack.pyD22-Oct-202512.7 KiB

test_pad_v3.pyD22-Oct-20256.5 KiB

test_paged_attention.pyD22-Oct-20255.1 KiB

test_paged_attention_mask.pyD22-Oct-20255.8 KiB

test_parallel_mirror_group.pyD22-Oct-20259.4 KiB

test_parallel_optimizer.pyD22-Oct-202516.9 KiB

test_parallel_optimizer_shared_cast.pyD22-Oct-20256.7 KiB

test_parallel_optimizer_without_grad.pyD22-Oct-20254.9 KiB

test_parallel_while.pyD22-Oct-20253.5 KiB

test_param_broadcast_cell.pyD22-Oct-20251.6 KiB

test_parameter_clone.pyD22-Oct-20251.3 KiB

test_parameter_init.pyD22-Oct-20251.9 KiB

test_parameter_merge.pyD22-Oct-20253.5 KiB

test_parameter_multi_users.pyD22-Oct-20253.6 KiB

test_pipeline_dynamic_shape.pyD22-Oct-20255.6 KiB

test_pipeline_end_node.pyD22-Oct-20253.2 KiB

test_pipeline_grad_freeze.pyD22-Oct-20254.7 KiB

test_pipeline_grad_reducer.pyD22-Oct-20255.2 KiB

test_pipeline_inference.pyD22-Oct-20258.5 KiB

test_pipeline_interleave.pyD22-Oct-20256.9 KiB

test_pipeline_interleave_multi_send_recv.pyD22-Oct-20254 KiB

test_pipeline_multiout.pyD22-Oct-20253.9 KiB

test_pipeline_opt_detection.pyD22-Oct-202514.7 KiB

test_pipeline_remove_node.pyD22-Oct-20254.8 KiB

test_pipeline_shard_with_mirror.pyD22-Oct-20254.4 KiB

test_pipeline_shared_parameter.pyD22-Oct-202516.9 KiB

test_pipeline_split.pyD22-Oct-202533 KiB

test_prelu.pyD22-Oct-20255.4 KiB

test_prelu_cell.pyD22-Oct-20252.8 KiB

test_print.pyD22-Oct-20256.1 KiB

test_prompt_flash_attention.pyD22-Oct-202514.4 KiB

test_prompt_k_v_cache.pyD22-Oct-20255.7 KiB

test_quant.pyD22-Oct-20253.4 KiB

test_quant_batch_matmul.pyD22-Oct-20258.8 KiB

test_quant_ops.pyD22-Oct-20256.4 KiB

test_quant_v2.pyD22-Oct-20254.4 KiB

test_random_choice_with_mask.pyD22-Oct-20252.3 KiB

test_range.pyD22-Oct-20253.5 KiB

test_range_dynamic.pyD22-Oct-20252.2 KiB

test_reduce_method_info.pyD22-Oct-202537.9 KiB

test_remove_cast_before_assign_add.pyD22-Oct-20251.9 KiB

test_repeated_calc.pyD22-Oct-20253.4 KiB

test_reshape.pyD22-Oct-202522.7 KiB

test_reshape_and_cache.pyD22-Oct-20255 KiB

test_reshape_consecutive.pyD22-Oct-20252.4 KiB

test_reshape_optimized.pyD22-Oct-20252 KiB

test_reshape_parameter.pyD22-Oct-20252.7 KiB

test_reshape_parameter_scalar_or_other_shape.pyD22-Oct-20256.6 KiB

test_reshape_shard_propagation.pyD22-Oct-202513.4 KiB

test_reshape_skip_redistribution.pyD22-Oct-20252.2 KiB

test_reshape_unexpand.pyD22-Oct-202510.8 KiB

test_resizebilinear.pyD22-Oct-202510.9 KiB

test_ring_attention.pyD22-Oct-20256.4 KiB

test_rms_norm.pyD22-Oct-20256.4 KiB

test_roi_align.pyD22-Oct-20255.1 KiB

test_scalar_loss.pyD22-Oct-20252.1 KiB

test_scatter_nd_ops.pyD22-Oct-202512.9 KiB

test_scatter_ops.pyD22-Oct-202510 KiB

test_scatter_update.pyD22-Oct-20253.3 KiB

test_select.pyD22-Oct-20253.9 KiB

test_self_attention.pyD22-Oct-20254.7 KiB

test_semi_auto_two_subgraphs.pyD22-Oct-20254.2 KiB

test_set_auto_parallel_context.pyD22-Oct-20256.1 KiB

test_set_offload_context.pyD22-Oct-20252 KiB

test_set_repeated_num_in_dev_matrix_right.pyD22-Oct-20252.5 KiB

test_set_stridedslice_strategy.pyD22-Oct-20252.6 KiB

test_sharding_propagation_for_conv2ds.pyD22-Oct-20256.6 KiB

test_shared_param_and_mix_precision.pyD22-Oct-20254.3 KiB

test_sigmoid_cross_entropy_with_logits.pyD22-Oct-20253.3 KiB

test_slice.pyD22-Oct-20256.3 KiB

test_softmax_cross_entropy_loss.pyD22-Oct-20254.1 KiB

test_softmax_gather_net.pyD22-Oct-202516 KiB

test_sparse_feature_bprop.pyD22-Oct-20254.5 KiB

test_sparse_gather_v2.pyD22-Oct-20257.2 KiB

test_split.pyD22-Oct-20255.4 KiB

test_split_ext.pyD22-Oct-20255.3 KiB

test_split_grad_sens.pyD22-Oct-20256.7 KiB

test_splitv.pyD22-Oct-20254.2 KiB

test_square.pyD22-Oct-20253.2 KiB

test_squeeze_info.pyD22-Oct-20252.8 KiB

test_stand_alone.pyD22-Oct-20253.5 KiB

test_step_parallel.pyD22-Oct-20252.7 KiB

test_strategy_checkpoint.pyD22-Oct-202513.7 KiB

test_stridedslice.pyD22-Oct-202527.1 KiB

test_sum_as_loss.pyD22-Oct-20252.8 KiB

test_syncbatchnorm.pyD22-Oct-20252.8 KiB

test_tile.pyD22-Oct-20255.8 KiB

test_topk.pyD22-Oct-20253.3 KiB

test_topk_ext.pyD22-Oct-20253.8 KiB

test_topkrouter.pyD22-Oct-20252.3 KiB

test_train_and_eval.pyD22-Oct-20252.5 KiB

test_transform_tensor.pyD22-Oct-202525.1 KiB

test_transpose.pyD22-Oct-20253.6 KiB

test_two_matmul.pyD22-Oct-202521.5 KiB

test_two_weights_parameter.pyD22-Oct-20252.9 KiB

test_uniform_candidate_sampler.pyD22-Oct-20257.1 KiB

test_uniform_real.pyD22-Oct-20252.5 KiB

test_unsortedsegmentmax.pyD22-Oct-20255.5 KiB

test_unsortedsegmentmin.pyD22-Oct-20255.5 KiB

test_unsortedsegmentprod.pyD22-Oct-20256.8 KiB

test_unsortedsegmentsum.pyD22-Oct-20258.1 KiB

test_using_seed_for_initializer.pyD22-Oct-20252.8 KiB

test_virtual_dataset_3_input.pyD22-Oct-20252.7 KiB

test_virtual_dataset_with_strategy.pyD22-Oct-202513.5 KiB

test_virtual_output.pyD22-Oct-202514.9 KiB

test_weight_quant_batch_matmul.pyD22-Oct-202514.5 KiB

test_wrong_parallel_mode_for_grad_reducer.pyD22-Oct-20254.8 KiB