• Home
Name Date Size #Lines LOC

..--

utils/06-May-2025-225178

__init__.pyD06-May-20251.5 KiB3921

conftest.pyD06-May-20251.2 KiB3114

test_adafactor.pyD06-May-20253.6 KiB10776

test_add_relu_redistribution.pyD06-May-20252.7 KiB9355

test_addn.pyD06-May-20252.4 KiB7849

test_allreduce_fusion.pyD06-May-202516 KiB405338

test_alltoall.pyD06-May-20259.4 KiB320238

test_anti_quant.pyD06-May-20253.7 KiB11369

test_arithmetic.pyD06-May-202546.9 KiB1,4021,108

test_asd.pyD06-May-20257.7 KiB234171

test_auto_parallel_BN_PReLU.pyD06-May-20252.4 KiB8750

test_auto_parallel_activation.pyD06-May-20252.1 KiB6639

test_auto_parallel_adasum.pyD06-May-20255.5 KiB146110

test_auto_parallel_arithmetic.pyD06-May-20256.9 KiB208155

test_auto_parallel_assign_sub_with_ref_key.pyD06-May-20252.4 KiB8148

test_auto_parallel_batch_matmul.pyD06-May-20252.9 KiB8861

test_auto_parallel_cast.pyD06-May-20253.3 KiB10164

test_auto_parallel_common_parameter.pyD06-May-20252.5 KiB8652

test_auto_parallel_data_parallel_mode.pyD06-May-20254.1 KiB12984

test_auto_parallel_double_sources.pyD06-May-20254.2 KiB13492

test_auto_parallel_double_star.pyD06-May-20253 KiB9762

test_auto_parallel_double_subgraphs.pyD06-May-20256.7 KiB195151

test_auto_parallel_fc_nobias.pyD06-May-20252.4 KiB8450

test_auto_parallel_flag.pyD06-May-20254.2 KiB11682

test_auto_parallel_for_loop.pyD06-May-20254.6 KiB144105

test_auto_parallel_for_loop_multi_subgraph.pyD06-May-20255.2 KiB149113

test_auto_parallel_for_loop_reshape.pyD06-May-20254.8 KiB148109

test_auto_parallel_for_loop_simplify.pyD06-May-20253.3 KiB10774

test_auto_parallel_four_matmul.pyD06-May-20254.5 KiB146100

test_auto_parallel_frontend_scheduling_with_pangu_alpha.pyD06-May-202531 KiB723639

test_auto_parallel_inference.pyD06-May-20252.1 KiB6236

test_auto_parallel_l2normalize.pyD06-May-20252.6 KiB8755

test_auto_parallel_matmul_drop.pyD06-May-20252.5 KiB8451

test_auto_parallel_matmul_prelu.pyD06-May-20252.9 KiB9560

test_auto_parallel_matmul_seq_parallel.pyD06-May-20255.9 KiB169127

test_auto_parallel_mem_coef_interface.pyD06-May-20251.6 KiB4928

test_auto_parallel_multi_graph.pyD06-May-20252.6 KiB7646

test_auto_parallel_onehot.pyD06-May-20254.6 KiB151105

test_auto_parallel_pangu_alpha.pyD06-May-202533.3 KiB774686

test_auto_parallel_parameter_cast.pyD06-May-20252.7 KiB8151

test_auto_parallel_partial_strategy.pyD06-May-20252.8 KiB9157

test_auto_parallel_rec_batch_gather.pyD06-May-20254 KiB12392

test_auto_parallel_rec_reducemin.pyD06-May-20252.4 KiB7953

test_auto_parallel_rec_reshape.pyD06-May-20255.3 KiB152115

test_auto_parallel_rec_softmax_loss.pyD06-May-20255.3 KiB148113

test_auto_parallel_rec_unknownop.pyD06-May-20256.7 KiB172147

test_auto_parallel_reduce_method.pyD06-May-20254.4 KiB148101

test_auto_parallel_reshape.pyD06-May-202513.6 KiB421327

test_auto_parallel_resnet_sharding_propagation.pyD06-May-202511.5 KiB319248

test_auto_parallel_resnet_sharding_propagation2.pyD06-May-202511.1 KiB315244

test_auto_parallel_rhombus.pyD06-May-20255.3 KiB163117

test_auto_parallel_sapp_RmsNorm.pyD06-May-20252.3 KiB6341

test_auto_parallel_sapp_batchparallel.pyD06-May-20251.8 KiB5531

test_auto_parallel_sapp_broadcast.pyD06-May-20251.8 KiB5633

test_auto_parallel_sapp_flash_attention.pyD06-May-20257.4 KiB212148

test_auto_parallel_sapp_gather.pyD06-May-20254.1 KiB11684

test_auto_parallel_sapp_opt_restrict.pyD06-May-20251.9 KiB5834

test_auto_parallel_sapp_strided_slice.pyD06-May-20252.8 KiB7250

test_auto_parallel_segment_min.pyD06-May-20252.4 KiB8148

test_auto_parallel_segment_sum.pyD06-May-20252.4 KiB8148

test_auto_parallel_shard_propagation.pyD06-May-20253.1 KiB9159

test_auto_parallel_shard_propagation2.pyD06-May-20252.2 KiB6438

test_auto_parallel_shard_propagation3.pyD06-May-20252.6 KiB7547

test_auto_parallel_share_param.pyD06-May-20254.4 KiB13497

test_auto_parallel_softmax_loss.pyD06-May-20252.4 KiB8048

test_auto_parallel_star_partial_strategy.pyD06-May-20255.7 KiB164127

test_auto_parallel_transpose.pyD06-May-20253.1 KiB9561

test_auto_parallel_triangle_overwrite.pyD06-May-20252.8 KiB8857

test_auto_parallel_tril.pyD06-May-20252.3 KiB7548

test_auto_parallel_tuple_depend.pyD06-May-20252.8 KiB9258

test_auto_parallel_two_bn.pyD06-May-20253 KiB10067

test_auto_parallel_two_matmul.pyD06-May-20257.6 KiB172132

test_auto_parallel_two_partial_matmul.pyD06-May-20252.5 KiB8551

test_auto_parallel_zig_zag.pyD06-May-20252.7 KiB9156

test_auto_pipeline.pyD06-May-20251.4 KiB4625

test_auto_star_elimination.pyD06-May-20253.2 KiB10265

test_autoparallel_rec_shared_param_strmodif.pyD06-May-20253.9 KiB10479

test_avgpool_3d.pyD06-May-20257.9 KiB195154

test_batch_matmul.pyD06-May-20254.9 KiB13194

test_batch_matmul_high_dimension.pyD06-May-202510.5 KiB346293

test_batch_parallel.pyD06-May-20254.4 KiB142102

test_batch_parallel_dropout.pyD06-May-20252.3 KiB7846

test_batch_parallel_tensoradd.pyD06-May-20252.2 KiB7644

test_batchmm.pyD06-May-20253.5 KiB9567

test_batchnorm.pyD06-May-20257.2 KiB215163

test_batchnorm_batch_parallel.pyD06-May-20254.8 KiB147103

test_batchparallel_replace_shape.pyD06-May-20252.9 KiB8859

test_bias_add.pyD06-May-20253.1 KiB10067

test_bias_add_comm_swap.pyD06-May-20254.1 KiB12789

test_bool_grad.pyD06-May-20252.4 KiB8250

test_bounding_box_encode.pyD06-May-20252.7 KiB8957

test_broadcast_dict.pyD06-May-20252.8 KiB8353

test_broadcast_to.pyD06-May-20254.2 KiB12386

test_cdist.pyD06-May-20254.7 KiB147105

test_cell_shard_check.pyD06-May-20257.2 KiB203153

test_checkvalid.pyD06-May-20254.2 KiB145106

test_combined_tensor.pyD06-May-20252.2 KiB5532

test_comm_fusion.pyD06-May-202510.5 KiB262205

test_comm_not_recompute.pyD06-May-20254.4 KiB11886

test_comparison_function_info.pyD06-May-202513.7 KiB413309

test_concat.pyD06-May-20256 KiB179129

test_config_layout.pyD06-May-20254.6 KiB10471

test_config_layout_for_arbitrary_ops.pyD06-May-20258.5 KiB235184

test_conformer.pyD06-May-202535.7 KiB819665

test_control_flow.pyD06-May-20253.2 KiB11078

test_conv2d.pyD06-May-202521.7 KiB553458

test_conv2d_transpose.pyD06-May-202513.5 KiB341276

test_conv3d.pyD06-May-20256.6 KiB163126

test_convolution.pyD06-May-20254.3 KiB11483

test_crop_and_resize.pyD06-May-20253.6 KiB10774

test_cum_ops.pyD06-May-20253.3 KiB9671

test_cus_matmul_dds.pyD06-May-20257.1 KiB201143

test_dataset.pyD06-May-20251.5 KiB4524

test_dataset_interface.pyD06-May-20255.8 KiB178126

test_decoder_k_v_cache.pyD06-May-20256.7 KiB12999

test_dense_matmul.pyD06-May-20252 KiB6037

test_dequant.pyD06-May-20251.9 KiB5831

test_device_num_not_power_of_2.pyD06-May-20257.5 KiB193151

test_different_type_for_div_op.pyD06-May-20253.6 KiB11474

test_distribute_predict.pyD06-May-20253.9 KiB11981

test_dropout.pyD06-May-20253.3 KiB10267

test_dropout_do_mask.pyD06-May-20254.4 KiB140103

test_dsd_matmul.pyD06-May-20257.2 KiB197131

test_dynamic_base_operation.pyD06-May-202521.8 KiB488384

test_dynamic_broadcast_to.pyD06-May-20252.9 KiB8355

test_dynamic_bs_data_parallel.pyD06-May-20253.3 KiB8258

test_dynamic_data_sink.pyD06-May-20258.7 KiB268217

test_dynamic_matmul.pyD06-May-20252 KiB5434

test_dynamic_pangu_alpha.pyD06-May-202571.4 KiB1,6251,347

test_dynamic_pangu_alpha_without_skip_redistribution.pyD06-May-202570.7 KiB1,6301,343

test_dynamic_quant.pyD06-May-20253.1 KiB9463

test_dynamic_shape.pyD06-May-20254.8 KiB13095

test_dynamic_shape_graph_utils.pyD06-May-202564.8 KiB1,3561,202

test_dynamic_shape_merge_allgather.pyD06-May-20254.8 KiB137103

test_dynamic_shape_multi_axis.pyD06-May-202525.1 KiB488427

test_dynamic_shape_no_redistribution.pyD06-May-202525.4 KiB732592

test_dynamic_stridedslice.pyD06-May-20253.3 KiB10268

test_element_wise_axis.pyD06-May-20252.1 KiB6439

test_element_wise_function.pyD06-May-202545 KiB1,2621,006

test_element_wise_three_inputs.pyD06-May-20252.2 KiB6439

test_element_wise_two_inputs.pyD06-May-20253.5 KiB9469

test_embeddinglookup.pyD06-May-20253.7 KiB11978

test_eval.pyD06-May-20253.1 KiB9462

test_expand_dims.pyD06-May-20254.1 KiB12383

test_ffn.pyD06-May-202511.3 KiB258210

test_ffn_quant.pyD06-May-20256.9 KiB138102

test_ffnext.pyD06-May-202511.3 KiB258210

test_ffnext_quant.pyD06-May-20256.9 KiB138102

test_fillv2.pyD06-May-20251.9 KiB5733

test_finegraint_micro_interleaved.pyD06-May-202520.2 KiB502420

test_flash_attention_score_info.pyD06-May-202521.6 KiB472411

test_fold_pipeline_shared_parameter.pyD06-May-20258.9 KiB231185

test_fold_pipeline_split.pyD06-May-202515.6 KiB387324

test_forward_graph.pyD06-May-20253.1 KiB9459

test_full_batch.pyD06-May-20253.9 KiB12183

test_func_mode.pyD06-May-20254.4 KiB12891

test_func_mode_pipeline.pyD06-May-20255.5 KiB161117

test_fused_infer_attention_score.pyD06-May-202518.5 KiB386340

test_gamma.pyD06-May-20252.9 KiB9059

test_gather_batch_dims.pyD06-May-20257.6 KiB208160

test_gather_field_split.pyD06-May-20254.8 KiB12586

test_gather_v2.pyD06-May-202519 KiB502412

test_gatherd.pyD06-May-20254.5 KiB13393

test_gathernd.pyD06-May-20253.5 KiB11579

test_gathernd_further.pyD06-May-202513 KiB392311

test_generate_batch_parallel.pyD06-May-20258.7 KiB263193

test_get_next.pyD06-May-20255.2 KiB152112

test_get_parameter_layout.pyD06-May-20252.6 KiB7344

test_global_norm.pyD06-May-20259.5 KiB225174

test_gpu_dropout.pyD06-May-20254 KiB12284

test_grad_accumulation.pyD06-May-20258.2 KiB198163

test_grid_sampler2d.pyD06-May-20254.6 KiB148104

test_group_norm.pyD06-May-20254 KiB10777

test_grouped_matmul.pyD06-May-202514 KiB381281

test_grouped_pairwise_exchange_alltoall.pyD06-May-20254.3 KiB12988

test_hybird_parallel_activation.pyD06-May-202510.2 KiB284218

test_incre_flash_attention.pyD06-May-20258.4 KiB260227

test_index_select.pyD06-May-20254.2 KiB12992

test_initializer_weight_slice.pyD06-May-20256.3 KiB165129

test_inplace_op.pyD06-May-20253.8 KiB11274

test_invert_populationcount.pyD06-May-20252.3 KiB7849

test_iou.pyD06-May-20252.5 KiB8554

test_kldiv_loss.pyD06-May-20254.9 KiB142105

test_kv_cache_scatter_update.pyD06-May-20253 KiB7851

test_l2_loss.pyD06-May-20252.5 KiB7949

test_l2normalize.pyD06-May-20252.6 KiB8351

test_layer_norm.pyD06-May-20254.5 KiB12083

test_layer_norm_further.pyD06-May-202515.7 KiB414327

test_layout_compare.pyD06-May-20253.6 KiB11984

test_layout_extend.pyD06-May-202515 KiB368314

test_layout_extend_activation.pyD06-May-20253.9 KiB12586

test_layout_extend_arithmetic.pyD06-May-20256.9 KiB189146

test_layout_extend_bias_add.pyD06-May-20254.1 KiB12289

test_layout_extend_layernorm.pyD06-May-20255.7 KiB153116

test_layout_extend_rmsnorm.pyD06-May-20255.1 KiB154112

test_layout_extend_transformer.pyD06-May-202513.7 KiB293248

test_layout_extend_transpose.pyD06-May-20255 KiB157114

test_lerp.pyD06-May-20256 KiB175139

test_lin_space.pyD06-May-20254.2 KiB13399

test_lin_space_ext.pyD06-May-20256.2 KiB177138

test_linear.pyD06-May-20252.6 KiB8150

test_load_tensor.pyD06-May-20251.4 KiB4421

test_loop_two_matmul.pyD06-May-20253 KiB10467

test_loss_and_o2_level.pyD06-May-20254.3 KiB13694

test_loss_and_optimizer.pyD06-May-20257.2 KiB219147

test_loss_scale.pyD06-May-20257.1 KiB204158

test_manual_embedding_lookup.pyD06-May-20257.7 KiB217162

test_manual_gatherv2.pyD06-May-20259.7 KiB260201

test_matmul_add_comm_reduction.pyD06-May-20259 KiB238182

test_matmul_dropout.pyD06-May-20254.5 KiB13796

test_matmul_tensor.pyD06-May-20255.3 KiB166114

test_matmul_transpose_attr.pyD06-May-202516 KiB401341

test_maxpool_3d.pyD06-May-20257.8 KiB194153

test_maxpool_avgpool.pyD06-May-20258.8 KiB209153

test_micro_batch_interleaved.pyD06-May-20252.5 KiB7546

test_mirror_insert.pyD06-May-20255.3 KiB154115

test_mix_precision_hybrid_parallel.pyD06-May-20253 KiB9560

test_model_flops_cal.pyD06-May-202515.2 KiB399321

test_model_with_loss.pyD06-May-20253.9 KiB12787

test_model_without_loss.pyD06-May-20256.5 KiB206153

test_moe_finalize_routing.pyD06-May-20253.2 KiB7540

test_mul_div_bn.pyD06-May-20252.9 KiB8252

test_mul_softmax_net.pyD06-May-20258.3 KiB237188

test_multi_field_embedding.pyD06-May-20255.5 KiB14998

test_neg.pyD06-May-20253.1 KiB9661

test_neighborexchange.pyD06-May-202518.2 KiB516400

test_neighborexchangev2.pyD06-May-202531.4 KiB794631

test_o2_level.pyD06-May-20257.1 KiB213160

test_one_dev.pyD06-May-20254.4 KiB13495

test_one_hot_net.pyD06-May-202513.7 KiB376302

test_one_weight_parameter.pyD06-May-20252.6 KiB8449

test_onehot.pyD06-May-20255.7 KiB180128

test_onehot_2dim.pyD06-May-20252.4 KiB7044

test_operator_model_parallel.pyD06-May-202512.8 KiB400294

test_optimizer.pyD06-May-20252.6 KiB7045

test_optimizer_clone_weight.pyD06-May-20253.6 KiB11469

test_pack.pyD06-May-202512.7 KiB356278

test_pad_v3.pyD06-May-20256.5 KiB186143

test_paged_attention.pyD06-May-20255.1 KiB11788

test_paged_attention_mask.pyD06-May-20255.8 KiB140109

test_parallel_mirror_group.pyD06-May-20259.4 KiB204166

test_parallel_optimizer.pyD06-May-202516.9 KiB427325

test_parallel_optimizer_shared_cast.pyD06-May-20256.7 KiB161124

test_parallel_optimizer_without_grad.pyD06-May-20254.9 KiB13897

test_parallel_while.pyD06-May-20253.5 KiB12287

test_param_broadcast_cell.pyD06-May-20251.6 KiB4424

test_parameter_clone.pyD06-May-20251.3 KiB4224

test_parameter_init.pyD06-May-20251.9 KiB6435

test_parameter_merge.pyD06-May-20253.5 KiB10573

test_parameter_multi_users.pyD06-May-20253.6 KiB11274

test_pipeline_dynamic_shape.pyD06-May-20255.6 KiB165125

test_pipeline_end_node.pyD06-May-20253.2 KiB10674

test_pipeline_grad_freeze.pyD06-May-20254.7 KiB11791

test_pipeline_grad_reducer.pyD06-May-20255.2 KiB169122

test_pipeline_inference.pyD06-May-20258.5 KiB234181

test_pipeline_interleave.pyD06-May-20256.9 KiB190149

test_pipeline_interleave_multi_send_recv.pyD06-May-20254 KiB10681

test_pipeline_multiout.pyD06-May-20253.9 KiB12292

test_pipeline_opt_detection.pyD06-May-202514.7 KiB377314

test_pipeline_remove_node.pyD06-May-20254.8 KiB142122

test_pipeline_shard_with_mirror.pyD06-May-20254.4 KiB13092

test_pipeline_shared_parameter.pyD06-May-202516.9 KiB463400

test_pipeline_split.pyD06-May-202533 KiB765658

test_prelu.pyD06-May-20255.4 KiB167117

test_prelu_cell.pyD06-May-20252.8 KiB9561

test_print.pyD06-May-20256.1 KiB175124

test_prompt_flash_attention.pyD06-May-202514.4 KiB315271

test_prompt_k_v_cache.pyD06-May-20255.7 KiB11183

test_quant.pyD06-May-20253.4 KiB11771

test_quant_batch_matmul.pyD06-May-20258.8 KiB232165

test_quant_ops.pyD06-May-20256.4 KiB200139

test_quant_v2.pyD06-May-20254.4 KiB13787

test_random_choice_with_mask.pyD06-May-20252.3 KiB7747

test_range.pyD06-May-20253.5 KiB11277

test_range_dynamic.pyD06-May-20252.2 KiB6741

test_reduce_method_info.pyD06-May-202537.9 KiB1,113884

test_remove_cast_before_assign_add.pyD06-May-20251.9 KiB5934

test_repeated_calc.pyD06-May-20253.4 KiB11069

test_reshape.pyD06-May-202522.7 KiB689518

test_reshape_and_cache.pyD06-May-20255 KiB11686

test_reshape_consecutive.pyD06-May-20252.4 KiB8154

test_reshape_optimized.pyD06-May-20252 KiB6437

test_reshape_parameter.pyD06-May-20252.7 KiB8954

test_reshape_parameter_scalar_or_other_shape.pyD06-May-20256.6 KiB180141

test_reshape_shard_propagation.pyD06-May-202513.4 KiB407319

test_reshape_skip_redistribution.pyD06-May-20252.2 KiB6336

test_reshape_unexpand.pyD06-May-202510.8 KiB326248

test_resizebilinear.pyD06-May-202510.9 KiB296235

test_ring_attention.pyD06-May-20256.4 KiB154123

test_rms_norm.pyD06-May-20256.4 KiB187142

test_roi_align.pyD06-May-20255.1 KiB148103

test_scalar_loss.pyD06-May-20252.1 KiB6839

test_scatter_nd_ops.pyD06-May-202512.9 KiB326258

test_scatter_ops.pyD06-May-202510 KiB269223

test_scatter_update.pyD06-May-20253.3 KiB9063

test_select.pyD06-May-20253.9 KiB12587

test_self_attention.pyD06-May-20254.7 KiB165108

test_semi_auto_two_subgraphs.pyD06-May-20254.2 KiB11784

test_set_auto_parallel_context.pyD06-May-20256.1 KiB143104

test_set_offload_context.pyD06-May-20252 KiB4931

test_set_repeated_num_in_dev_matrix_right.pyD06-May-20252.5 KiB6541

test_set_stridedslice_strategy.pyD06-May-20252.6 KiB7550

test_sharding_propagation_for_conv2ds.pyD06-May-20256.6 KiB159126

test_shared_param_and_mix_precision.pyD06-May-20254.3 KiB10766

test_sigmoid_cross_entropy_with_logits.pyD06-May-20253.3 KiB9661

test_slice.pyD06-May-20256.3 KiB179130

test_softmax_cross_entropy_loss.pyD06-May-20254.1 KiB12885

test_softmax_gather_net.pyD06-May-202516 KiB426351

test_sparse_feature_bprop.pyD06-May-20254.5 KiB13489

test_sparse_gather_v2.pyD06-May-20257.2 KiB207154

test_split.pyD06-May-20255.4 KiB159111

test_split_ext.pyD06-May-20255.3 KiB160116

test_split_grad_sens.pyD06-May-20256.7 KiB210151

test_splitv.pyD06-May-20254.2 KiB12588

test_square.pyD06-May-20253.2 KiB9863

test_squeeze_info.pyD06-May-20252.8 KiB9257

test_stand_alone.pyD06-May-20253.5 KiB12779

test_step_parallel.pyD06-May-20252.7 KiB8453

test_strategy_checkpoint.pyD06-May-202513.7 KiB338274

test_stridedslice.pyD06-May-202527.1 KiB645532

test_sum_as_loss.pyD06-May-20252.8 KiB9156

test_syncbatchnorm.pyD06-May-20252.8 KiB7348

test_tile.pyD06-May-20255.8 KiB173122

test_topk.pyD06-May-20253.3 KiB11478

test_topk_ext.pyD06-May-20253.8 KiB13195

test_topkrouter.pyD06-May-20252.3 KiB7147

test_train_and_eval.pyD06-May-20252.5 KiB7648

test_transform_tensor.pyD06-May-202525.1 KiB560494

test_transpose.pyD06-May-20253.6 KiB11779

test_two_matmul.pyD06-May-202521.5 KiB601466

test_two_weights_parameter.pyD06-May-20252.9 KiB8854

test_uniform_candidate_sampler.pyD06-May-20257.1 KiB183136

test_uniform_real.pyD06-May-20252.5 KiB8151

test_unsortedsegmentmax.pyD06-May-20255.5 KiB168117

test_unsortedsegmentmin.pyD06-May-20255.5 KiB168117

test_unsortedsegmentprod.pyD06-May-20256.8 KiB195147

test_unsortedsegmentsum.pyD06-May-20258.1 KiB228175

test_using_seed_for_initializer.pyD06-May-20252.8 KiB7247

test_virtual_dataset_3_input.pyD06-May-20252.7 KiB8653

test_virtual_dataset_with_strategy.pyD06-May-202513.5 KiB350286

test_virtual_output.pyD06-May-202514.9 KiB412339

test_weight_quant_batch_matmul.pyD06-May-202514.5 KiB296223

test_wrong_parallel_mode_for_grad_reducer.pyD06-May-20254.8 KiB138104