mindspore 2.5.0__cp311-cp311-win_amd64.whl → 2.6.0rc1__cp311-cp311-win_amd64.whl

mindspore/ops/auto_generate/cpp_create_prim_instance_helper.py

@@ -17,14 +17,17 @@
 from mindspore.common import dtype as mstype
 
 op_args_default_value = {
-    "AdamWeightDecay": {"use_locking": False},
     "AdamW": {"amsgrad": False, "maximize": False},
-    "AddExt": {"alpha": 1},
-    "AddLayerNormV2": {"epsilon": 1e-5, "additionalOut": False},
-    "AddRmsNorm": {"epsilon": 1e-6},
+    "AdamWeightDecay": {"use_locking": False},
     "Addbmm": {"beta": 1, "alpha": 1},
+    "AddcdivExt": {"value": 1},
+    "AddcmulExt": {"value": 1},
     "Addmm": {"beta": 1, "alpha": 1},
     "Addmv": {"beta": 1, "alpha": 1},
+    "AddExt": {"alpha": 1},
+    "AddLayerNormV2": {"epsilon": 1e-5, "additionalOut": False},
+    "AddRmsNorm": {"epsilon": 1e-6},
+    "AddScalar": {"alpha": 1},
     "AllGatherMatmul": {"bias": None, "gather_index": 0, "gather_output": True, "comm_turn": 0, "trans_input": False, "trans_x2": False},
     "ApplyAdamW": {"max_grad_norm": None, "amsgrad": False, "maximize": False},
     "ApplyCamePart2": {"sum_r": None, "global_shape": None},
@@ -38,13 +41,16 @@ op_args_default_value = {
     "ArgMinExt": {"dim": None, "keepdim": False},
     "Argmin": {"axis": -1, "output_type": mstype.int32},
     "ArgMinWithValue": {"axis": 0, "keep_dims": False},
-    "ArgSort": {"dim": -1, "descending": False},
+    "ArgSort": {"dim": -1, "descending": False, "stable": False},
     "AsStrided": {"storage_offset": 0},
     "AvgPool1D": {"stride": None, "padding": 0, "ceil_mode": False, "count_include_pad": True},
     "AvgPool2DGrad": {"padding": 0, "ceil_mode": False, "count_include_pad": True, "divisor_override": None},
     "AvgPool2D": {"padding": 0, "ceil_mode": False, "count_include_pad": True, "divisor_override": None},
+    "AvgPool3DExt": {"stride": None, "padding": 0, "ceil_mode": False, "count_include_pad": True, "divisor_override": None},
+    "AvgPool3DGradExt": {"stride": None, "padding": 0, "ceil_mode": False, "count_include_pad": True, "divisor_override": None},
     "AvgPoolGrad": {"kernel_size": 1, "strides": 1, "pad_mode": 'VALID', "data_format": 'NCHW'},
     "AvgPool": {"kernel_size": 1, "strides": 1, "pad_mode": 'VALID', "data_format": 'NCHW'},
+    "Baddbmm": {"beta": 1, "alpha": 1},
     "BatchMatMul": {"transpose_a": False, "transpose_b": False},
     "BatchNormElemt": {"weight": None, "bias": None, "mean": None, "invstd": None, "eps": 1e-5},
     "BatchNormExt": {"running_mean": None, "runnning_var": None, "training": False, "momentum": 0.1, "epsilon": 1e-5},
@@ -53,6 +59,7 @@ op_args_default_value = {
     "BatchNormGradGrad": {"is_training": False, "epsilon": 1e-5, "data_format": 'NCHW'},
     "BatchNormGrad": {"is_training": False, "epsilon": 1e-5, "data_format": 'NCHW'},
     "BatchNormGradWithActivation": {"is_training": False, "epsilon": 1e-5, "data_format": 'NCHW'},
+    "BatchNormGradWithAddAndActivation": {"is_training": False, "epsilon": 1e-5, "data_format": 'NCHW'},
     "BatchNorm": {"is_training": False, "epsilon": 1e-5, "momentum": 0.1, "data_format": 'NCHW'},
     "BatchNormWithActivation": {"is_training": False, "epsilon": 1e-5, "momentum": 0.1, "data_format": 'NCHW'},
     "BatchNormWithAddAndActivation": {"is_training": False, "epsilon": 1e-5, "momentum": 0.1, "data_format": 'NCHW'},
@@ -75,35 +82,38 @@ op_args_default_value = {
     "Col2ImGrad": {"dilation": 1, "padding": 0, "stride": 1},
     "Concat": {"axis": 0},
     "ConstantPadND": {"value": 0.0},
+    "Conv1DExt": {"bias": None, "stride": 1, "padding": 0, "dilation": 1, "groups": 1},
+    "Conv1DPadding": {"bias": None, "stride": 1, "padding": 'valid', "dilation": 1, "groups": 1},
     "Conv2DExt": {"bias": None, "stride": 1, "padding": 0, "dilation": 1, "groups": 1},
     "Conv2DPadding": {"bias": None, "stride": 1, "padding": 'valid', "dilation": 1, "groups": 1},
     "Conv3DExt": {"bias": None, "stride": 1, "padding": 0, "dilation": 1, "groups": 1},
     "Conv3DPadding": {"bias": None, "stride": 1, "padding": 'valid', "dilation": 1, "groups": 1},
-    "ConvTranspose2D": {"bias": None, "stride": 1, "padding": 0, "output_padding": 0, "groups": 1, "dilation": 1},
     "ConvolutionGrad": {"bias": None, "stride": 1, "padding": 0, "dilation": (1, 1), "transposed": False, "output_padding": 0, "groups": 1, "output_mask": ()},
     "Convolution": {"bias": None, "stride": 1, "padding": 0, "dilation": 1, "transposed": False, "output_padding": 0, "groups": 1},
     "ConvolutionStrGrad": {"bias": None, "stride": 1, "padding": 'valid', "dilation": 1, "transposed": False, "output_padding": 0, "groups": 1, "output_mask": ()},
     "ConvolutionStr": {"bias": None, "stride": 1, "padding": 'valid', "dilation": 1, "transposed": False, "output_padding": 0, "groups": 1},
-    "Correlate": {"mode": 'valid'},
+    "ConvTranspose2D": {"bias": None, "stride": 1, "padding": 0, "output_padding": 0, "groups": 1, "dilation": 1},
+    "Correlate": {"pad_mode": 'valid'},
     "CountNonZero": {"dim": None},
     "Cross": {"dim": -65530},
+    "CumsumExt": {"dtype": None},
     "CumProd": {"exclusive": False, "reverse": False},
     "CumSum": {"exclusive": False, "reverse": False},
-    "CumsumExt": {"dtype": None},
-    "DCT": {"type": 2, "n": None, "axis": -1, "norm": None},
     "DCTN": {"type": 2, "s": None, "axes": None, "norm": None},
+    "DCT": {"type": 2, "n": None, "axis": -1, "norm": None},
     "Dense": {"bias": None},
     "Diagonal": {"offset": 0, "dim1": 0, "dim2": 1},
-    "DivMod": {"rounding_mode": None},
+    "DiagExt": {"diagonal": 0},
     "DivMods": {"rounding_mode": None},
+    "DivMod": {"rounding_mode": None},
     "Dropout": {"keep_prob": 0.5, "Seed0": 0, "Seed1": 0},
     "Eig": {"compute_v": False},
     "EluExt": {"alpha": 1.0},
-    "EluGradExt": {"alpha": 1.0},
+    "EluGradExt": {"alpha": 1.0, "is_result": False},
     "Elu": {"alpha": 1.0},
-    "EmbeddingApplyAdaGrad": {"mask_zero": (0,), "padding_key": (0,), "padding_key_mask": (1,), "completion_key": (0,), "completion_key_mask": (1,), "_embedding_dim": 1, "_max_key_num": 1},
-    "EmbeddingApplyAdam": {"mask_zero": (0,), "padding_key": (0,), "padding_key_mask": (1,), "completion_key": (0,), "completion_key_mask": (1,), "_embedding_dim": 1, "_max_key_num": 1},
     "EmbeddingApplyAdamW": {"ams_grad": (0,), "mask_zero": (0,), "padding_key": (0,), "padding_key_mask": (1,), "completion_key": (0,), "completion_key_mask": (1,), "_embedding_dim": 1, "_max_key_num": 1},
+    "EmbeddingApplyAdam": {"mask_zero": (0,), "padding_key": (0,), "padding_key_mask": (1,), "completion_key": (0,), "completion_key_mask": (1,), "_embedding_dim": 1, "_max_key_num": 1},
+    "EmbeddingApplyAdaGrad": {"mask_zero": (0,), "padding_key": (0,), "padding_key_mask": (1,), "completion_key": (0,), "completion_key_mask": (1,), "_embedding_dim": 1, "_max_key_num": 1},
     "EmbeddingApplyFtrl": {"mask_zero": (0,), "padding_key": (0,), "padding_key_mask": (1,), "completion_key": (0,), "completion_key_mask": (1,), "_embedding_dim": 1, "_max_key_num": 1},
     "EmbeddingApplyRmsprop": {"mask_zero": (0,), "padding_key": (0,), "padding_key_mask": (1,), "completion_key": (0,), "completion_key_mask": (1,), "_embedding_dim": 1, "_max_key_num": 1},
     "EmbeddingApplySgd": {"mask_zero": (0,), "padding_key": (0,), "padding_key_mask": (1,), "completion_key": (0,), "completion_key_mask": (1,), "_embedding_dim": 1, "_max_key_num": 1},
@@ -118,12 +128,12 @@ op_args_default_value = {
     "ExtractImagePatches": {"padding": 'VALID'},
     "FFNExt": {"expertTokens": None, "bias1": None, "bias2": None, "scale": None, "offset": None, "deqScale1": None, "deqScale2": None, "antiquant_scale1": None, "antiquant_scale2": None, "antiquant_offset1": None, "antiquant_offset2": None, "activation": 'fastgelu', "inner_precise": 0},
     "FFT2": {"s": None, "dim": (-2, -1), "norm": None},
-    "FFT": {"n": None, "dim": -1, "norm": None},
-    "FFTOrtho": {"axes": None, "forward": True},
-    "FFTWithSize": {"norm": 'backward', "onesided": True, "signal_sizes": ()},
     "FFTFreq": {"d": 1.0, "dtype": None},
     "FFTN": {"s": None, "dim": None, "norm": None},
     "FFTShift": {"dim": None},
+    "FFT": {"n": None, "dim": -1, "norm": None},
+    "FFTOrtho": {"axes": None, "forward": True},
+    "FFTWithSize": {"norm": 'backward', "onesided": True, "signal_sizes": ()},
     "FillScalar": {"dtype": None},
     "FillTensor": {"dtype": None},
     "FlashAttentionScoreGrad": {"pse_shift": None, "drop_mask": None, "padding_mask": None, "atten_mask": None, "softmax_max": None, "softmax_sum": None, "softmax_in": None, "attention_in": None, "prefix": None, "actual_seq_qlen": None, "actual_seq_kvlen": None, "keep_prob": 1.0, "scale_value": 1.0, "pre_tokens": 65536, "next_tokens": 65536, "inner_precise": 1, "input_layout": 'BSH', "sparse_mode": 0},
@@ -131,6 +141,8 @@ op_args_default_value = {
     "FlattenExt": {"start_dim": 0, "end_dim": -1},
     "FullLike": {"dtype": None},
     "Gather": {"batch_dims": 0},
+    "GeluExt": {"approximate": 'none'},
+    "GeluGradExt": {"approximate": 'none'},
     "GenerateEodMaskV2": {"start": 0, "steps": 1, "error_mode": 'cycle', "flip_mode": 'bitflip', "multiply_factor": 0.0, "bit_pos": 0, "flip_probability": 0.0},
     "GLU": {"axis": -1},
     "GmmBackward": {"group_list": None},
@@ -141,46 +153,53 @@ op_args_default_value = {
     "GridSampler3D": {"interpolation_mode": 'bilinear', "padding_mode": 'zeros', "align_corners": False},
     "GroupNormGrad": {"dx_is_require": True, "dgamma_is_require": True, "dbeta_is_require": True},
     "GroupNorm": {"weight": None, "bias": None, "eps": 1e-5},
+    "GroupTopk": {"k_inner": 1},
     "HardtanhGrad": {"min_val": -1, "max_val": 1},
     "Hardtanh": {"min_val": -1, "max_val": 1},
     "HFFT2": {"s": None, "dim": (-2, -1), "norm": None},
-    "HFFT": {"n": None, "dim": -1, "norm": None},
     "HFFTN": {"s": None, "dim": None, "norm": None},
+    "HFFT": {"n": None, "dim": -1, "norm": None},
     "HistcExt": {"bins": 100, "min": 0, "max": 0},
     "HShrinkGrad": {"lambd": 0.5},
     "HShrink": {"lambd": 0.5},
-    "IDCT": {"type": 2, "n": None, "axis": -1, "norm": None},
     "IDCTN": {"type": 2, "s": None, "axes": None, "norm": None},
+    "IDCT": {"type": 2, "n": None, "axis": -1, "norm": None},
     "IFFT2": {"s": None, "dim": (-2, -1), "norm": None},
-    "IFFT": {"n": None, "dim": -1, "norm": None},
     "IFFTN": {"s": None, "dim": None, "norm": None},
     "IFFTShift": {"dim": None},
+    "IFFT": {"n": None, "dim": -1, "norm": None},
     "IHFFT2": {"s": None, "dim": (-2, -1), "norm": None},
-    "IHFFT": {"n": None, "dim": -1, "norm": None},
     "IHFFTN": {"s": None, "dim": None, "norm": None},
+    "IHFFT": {"n": None, "dim": -1, "norm": None},
     "Im2ColExt": {"dilation": 1, "padding": 0, "stride": 1},
     "IncreFlashAttention": {"attn_mask": None, "actual_seq_lengths": None, "pse_shift": None, "dequant_scale1": None, "quant_scale1": None, "dequant_scale2": None, "quant_scale2": None, "quant_offset2": None, "antiquant_scale": None, "antiquant_offset": None, "block_table": None, "kv_padding_size": None, "num_heads": 1, "input_layout": 'BSH', "scale_value": 1.0, "num_key_value_heads": 0, "block_size": 0, "inner_precise": 1},
     "IndexAddExt": {"alpha": 1},
     "InnerInplaceIndexPut": {"accumulate": False},
-    "InplaceAddExt": {"alpha": 1},
     "InplaceAddmm": {"beta": 1, "alpha": 1},
     "InplaceAddsExt": {"alpha": 1},
+    "InplaceAddExt": {"alpha": 1},
     "InplaceClampScalar": {"min": None, "max": None},
     "InplaceClampTensor": {"min": None, "max": None},
-    "InplaceDivMod": {"rounding_mode": None},
     "InplaceDivMods": {"rounding_mode": None},
+    "InplaceDivMod": {"rounding_mode": None},
+    "InplaceElu": {"alpha": 1.0},
+    "InplaceFillDiagonal": {"wrap": False},
     "InplaceHardtanh": {"min_val": -1, "max_val": 1},
     "InplaceIndexAddExt": {"alpha": 1},
     "InplaceIndexPut": {"accumulate": False},
+    "InplacePut": {"accumulate": False},
     "InplaceRandom": {"from_": 0, "to": None, "seed": 0, "offset": 0},
     "InplaceSubExt": {"alpha": 1},
     "InplaceSubScalar": {"alpha": 1},
     "InsertGemV2InBackward": {"start": 0, "steps": 1, "error_mode": 'cycle', "flip_mode": 'bitflip', "multiply_factor": 0.0, "bit_pos": 0, "flip_probability": 0.0},
     "IRFFT2": {"s": None, "dim": (-2, -1), "norm": None},
+    "IRFFTN": {"s": None, "dim": None, "norm": None},
     "IRFFTDouble": {"dim": -1},
     "IRFFT": {"n": None, "dim": -1, "norm": None},
-    "IRFFTN": {"s": None, "dim": None, "norm": None},
     "IsClose": {"rtol": 1e-05, "atol": 1e-08, "equal_nan": False},
+    "KLDivGrad": {"reduction": 'mean', "log_target": False},
+    "KLDiv": {"reduction": 'mean', "log_target": False},
+    "Kthvalue": {"dim": -1, "keepdim": False},
     "L1LossBackwardExt": {"reduction": 'mean'},
     "L1LossExt": {"reduction": 'mean'},
     "LayerNormExt": {"weight": None, "bias": None, "eps": 1e-5},
@@ -192,35 +211,40 @@ op_args_default_value = {
     "LayerNormV3": {"begin_norm_axis": 1, "begin_params_axis": 1, "epsilon": 1e-7},
     "LeakyReLUExt": {"negative_slope": 0.01},
     "LeakyReLUGradExt": {"negative_slope": 0.01, "is_result": False},
-    "LinSpaceExt": {"dtype": None},
+    "LinalgQr": {"mode": 'reduced'},
     "LinalgVectorNorm": {"ord": 2, "dim": None, "keepdim": False, "dtype": None},
-    "LogSoftmaxExt": {"dim": None, "dtype": None},
-    "LogSoftmaxGrad": {"axis": -1},
-    "LogSoftmax": {"axis": -1},
+    "LinSpaceExt": {"dtype": None},
     "LogitGrad": {"eps": -1.0},
     "Logit": {"eps": -1.0},
     "LogSumExp": {"keepdim": False},
+    "LogSoftmaxExt": {"dim": None, "dtype": None},
+    "LogSoftmaxGrad": {"axis": -1},
+    "LogSoftmax": {"axis": -1},
     "LpNormV2": {"p": 2, "dim": None, "keepdim": False, "epsilon": 1e-12},
     "LstsqV2": {"driver": None},
     "MatMul": {"transpose_a": False, "transpose_b": False},
     "MatmulReduceScatter": {"reduce_op": 'sum', "bias": None, "comm_turn": 0, "trans_input": False, "trans_x2": False},
+    "MaximumGradGrad": {"grad_x": True, "grad_y": True},
+    "MaximumGrad": {"grad_x": True, "grad_y": True},
     "MaxDim": {"keepdim": False},
     "MaxPoolGradWithIndices": {"strides": None, "pads": 0, "dilation": (1, 1), "ceil_mode": False, "argmax_type": mstype.int64},
     "MaxPoolGradWithMask": {"strides": None, "pads": 0, "dilation": (1, 1), "ceil_mode": False, "argmax_type": mstype.int64},
     "MaxPoolWithIndices": {"strides": None, "pads": 0, "dilation": (1, 1), "ceil_mode": False, "argmax_type": mstype.int64},
     "MaxPoolWithMask": {"strides": None, "pads": 0, "dilation": (1, 1), "ceil_mode": False, "argmax_type": mstype.int64},
     "MaxUnpool2DExt": {"stride": None, "padding": 0, "output_size": None},
-    "MaximumGradGrad": {"grad_x": True, "grad_y": True},
-    "MaximumGrad": {"grad_x": True, "grad_y": True},
     "MeanExt": {"dim": None, "keepdim": False, "dtype": None},
     "MedianDim": {"dim": -1, "keepdim": False},
     "Meshgrid": {"indexing": 'xy'},
-    "MinDim": {"keepdim": False},
     "MinimumGrad": {"grad_x": True, "grad_y": True},
+    "MinDim": {"keepdim": False},
+    "MoeTokenPermuteGrad": {"num_topk": 1, "padded_mode": False},
+    "MoeTokenPermute": {"num_out_tokens": None, "padded_mode": False},
+    "MoeTokenUnpermuteGrad": {"probs": None, "padded_mode": False, "restore_shape": None},
+    "MoeTokenUnpermute": {"probs": None, "padded_mode": False, "restore_shape": None},
     "MSELossExt": {"reduction": 'mean'},
     "MSELossGradExt": {"reduction": 'mean'},
-    "NanToNum": {"nan": None, "posinf": None, "neginf": None},
     "Nansum": {"dim": None, "keepdim": False, "dtype": None},
+    "NanToNum": {"nan": None, "posinf": None, "neginf": None},
     "NewEmpty": {"dtype": None, "device": None},
     "NewOnes": {"dtype": None},
     "NewZeros": {"dtype": None},
@@ -228,24 +252,24 @@ op_args_default_value = {
     "NLLLossGrad": {"reduction": 'mean', "ignore_index": -100},
     "NLLLoss": {"reduction": 'mean', "ignore_index": -100},
     "Norm": {"p": 2.0, "dim": None, "keepdim": False, "dtype": None},
-    "OneHotExt": {"axis": -1},
-    "OneHot": {"axis": -1},
     "OnesLikeExt": {"dtype": None},
     "Ones": {"dtype": None},
+    "OneHotExt": {"axis": -1},
+    "OneHot": {"axis": -1},
     "PagedAttentionMask": {"antiquant_scale": None, "antiquant_offset": None, "alibi_mask": None, "kv_cache_quant_mode": 'DEFAULT'},
-    "PagedAttention": {"antiquant_scale": None, "antiquant_offset": None, "attn_mask": None, "q_seq_lens": None, "kv_cache_quant_mode": 'DEFAULT'},
+    "PagedAttention": {"antiquant_scale": None, "antiquant_offset": None, "attn_mask": None, "q_seq_lens": None, "alibi_mask": None, "kv_cache_quant_mode": 'DEFAULT', "mask_mode": 'MASK_DEFAULT', "mla_v_dim": 0},
     "ProdExt": {"dim": None, "keepdim": False, "dtype": None},
     "PromptFlashAttention": {"attn_mask": None, "actual_seq_lengths": None, "actual_seq_lengths_kv": None, "pse_shift": None, "deq_scale1": None, "quant_scale1": None, "deq_scale2": None, "quant_scale2": None, "quant_offset2": None, "num_heads": 1, "scale_value": 1.0, "pre_tokens": 2147483647, "next_tokens": 0, "input_layout": 'BSH', "num_key_value_heads": 0, "sparse_mode": 0, "inner_precise": 1},
     "PromptKVCache": {"align_mode": 'LEFT'},
     "Qr": {"full_matrices": False},
-    "RandExt": {"dtype": None},
-    "RandLikeExt": {"dtype": None},
     "RandIntLike": {"dtype": None},
     "RandInt": {"dtype": None},
     "RandnLike": {"dtype": None},
     "Randn": {"dtype": None},
     "RandpermExt": {"dtype": mstype.int64},
     "RandpermV2": {"seed": 0, "offset": 0, "dtype": mstype.int64},
+    "RandExt": {"dtype": None},
+    "RandLikeExt": {"dtype": None},
     "Range": {"maxlen": 1000000},
     "ReduceAll": {"axis": None, "keep_dims": False},
     "ReduceAny": {"axis": (), "keep_dims": False},
@@ -257,6 +281,7 @@ op_args_default_value = {
     "ReduceSum": {"axis": (), "keep_dims": False, "skip_mode": False},
     "RepeatInterleaveInt": {"dim": None, "output_size": None},
     "RepeatInterleaveTensor": {"dim": None, "output_size": None},
+    "ReshapeAndCache": {"value": None, "key_cache": None, "value_cache": None, "slot_mapping": None},
     "ResizeBicubicGrad": {"align_corners": False, "half_pixel_centers": False},
     "ResizeBicubic": {"align_corners": False, "half_pixel_centers": False},
     "ResizeBilinearGrad": {"align_corners": False, "half_pixel_centers": False},
@@ -269,11 +294,11 @@ op_args_default_value = {
     "ResizeNearestNeighborV2Grad": {"align_corners": False, "half_pixel_centers": False},
     "ResizeNearestNeighborV2": {"align_corners": False, "half_pixel_centers": False},
     "RFFT2": {"s": None, "dim": (-2, -1), "norm": None},
-    "RFFT": {"n": None, "dim": -1, "norm": None},
     "RFFTFreq": {"d": 1.0, "dtype": None},
     "RFFTN": {"s": None, "dim": None, "norm": None},
+    "RFFT": {"n": None, "dim": -1, "norm": None},
     "RmsNorm": {"epsilon": 1e-6},
-    "Roll": {"axis": None},
+    "Roll": {"dims": None},
     "RotaryPositionEmbeddingGrad": {"dx": None, "mode": 0},
     "RotaryPositionEmbedding": {"mode": 0},
     "RotatedIou": {"trans": False, "mode": 0, "is_cross": True, "v_threshold": 0.0, "e_threshold": 0.0},
@@ -293,8 +318,12 @@ op_args_default_value = {
     "SoftplusGradExt": {"beta": 1, "threshold": 20},
     "SoftShrinkGrad": {"lambd": 0.5},
     "SoftShrink": {"lambd": 0.5},
+    "SoftMarginLossGrad": {"reduction": 'mean'},
+    "SoftMarginLoss": {"reduction": 'mean'},
     "SolveTriangular": {"trans": 0, "lower": False, "unit_diagonal": False},
     "SortExt": {"dim": -1, "descending": False, "stable": False},
+    "SpeedFusionAttentionGrad": {"pse": None, "padding_mask": None, "atten_mask": None, "softmax_max": None, "softmax_sum": None, "softmax_in": None, "attention_in": None, "scale_value": 1.0, "keep_prob": 1.0, "pre_tokens": 2147483647, "next_tokens": 2147483647, "inner_precise": 0, "seed": None, "offset": None, "numels": None, "prefix": None, "actual_seq_qlen": None, "actual_seq_kvlen": None, "sparse_mode": 0, "gen_mask_parallel": True, "sync": False, "pse_type": 1, "q_start_idx": None, "kv_start_idx": None},
+    "SpeedFusionAttention": {"pse": None, "padding_mask": None, "atten_mask": None, "scale": 1.0, "keep_prob": 1.0, "pre_tokens": 2147483647, "next_tokens": 2147483647, "inner_precise": 0, "prefix": None, "actual_seq_qlen": None, "actual_seq_kvlen": None, "sparse_mode": 0, "gen_mask_parallel": True, "sync": False, "pse_type": 1, "q_start_idx": None, "kv_start_idx": None},
     "Split": {"axis": 0, "output_num": 1},
     "SplitTensor": {"dim": 0},
     "SplitWithSize": {"dim": 0},
@@ -304,19 +333,22 @@ op_args_default_value = {
     "Std": {"dim": None, "correction": 1, "keepdim": False},
     "StridedSlice": {"begin_mask": 0, "end_mask": 0, "ellipsis_mask": 0, "new_axis_mask": 0, "shrink_axis_mask": 0},
     "SubExt": {"alpha": 1},
+    "SubScalar": {"alpha": 1},
     "SumExt": {"dim": None, "keepdim": False, "dtype": None},
+    "Svd": {"full_matrices": False, "compute_uv": True},
     "SwigluGrad": {"dim": -1},
     "Swiglu": {"dim": -1},
     "TensorScatterElements": {"axis": 0, "reduce": 'none'},
-    "TopkExt": {"dim": -1, "largest": True, "sorted": True},
     "TopKRouter": {"drop_type": 0},
+    "TopkExt": {"dim": -1, "largest": True, "sorted": True},
     "TraceV2Grad": {"offset": 0, "axis1": 1, "axis2": 0},
     "TraceV2": {"offset": 0, "axis1": 1, "axis2": 0, "dtype": None},
+    "TriangularSolve": {"upper": True, "transpose": False, "unitriangular": False},
     "TrilExt": {"diagonal": 0},
     "Triu": {"diagonal": 0},
     "TupleToTensor": {"dtype": None},
     "Unique2": {"sorted": True, "return_inverse": False, "return_counts": False},
-    "UniqueConsecutive": {"return_idx": False, "return_counts": False, "axis": None},
+    "UniqueConsecutive": {"return_inverse": False, "return_counts": False, "dim": None},
     "UnstackExt": {"dim": 0},
     "UpsampleBicubic2DGrad": {"output_size": None, "scales": None, "align_corners": False},
     "UpsampleBicubic2D": {"output_size": None, "scales": None, "align_corners": False},
@@ -339,8 +371,8 @@ op_args_default_value = {
     "AddRmsNormDynamicQuant": {"smooth_scale2": None, "epsilon": 1e-5},
     "AddRmsNormQuantV2": {"epsilon": 1e-5},
     "DynamicQuantExt": {"smooth_scales": None},
-    "FusedInferAttentionScore": {"pse_shift": None, "attn_mask": None, "actual_seq_lengths": None, "actual_seq_lengths_kv": None, "dequant_scale1": None, "quant_scale1": None, "dequant_scale2": None, "quant_scale2": None, "quant_offset2": None, "antiquant_scale": None, "antiquant_offset": None, "block_table": None, "query_padding_size": None, "kv_padding_size": None, "scale_value": 1.0, "pre_tokens": 2147483647, "next_tokens": 2147483647, "input_layout": 'BSH', "num_key_value_heads": 0, "sparse_mode": 0, "inner_precise": 1, "block_size": 0, "antiquant_mode": 0, "softmax_lse_flag": False},
-    "GroupedMatmul": {"bias": None, "scale": None, "offset": None, "antiquant_scale": None, "antiquant_offset": None, "group_list": None, "split_item": 0, "group_type": -1},
+    "FusedInferAttentionScore": {"pse_shift": None, "attn_mask": None, "actual_seq_lengths": None, "actual_seq_lengths_kv": None, "dequant_scale1": None, "quant_scale1": None, "dequant_scale2": None, "quant_scale2": None, "quant_offset2": None, "antiquant_scale": None, "antiquant_offset": None, "block_table": None, "query_padding_size": None, "kv_padding_size": None, "key_antiquant_scale": None, "key_antiquant_offset": None, "value_antiquant_scale": None, "value_antiquant_offset": None, "key_shared_prefix": None, "value_shared_prefix": None, "actual_shared_prefix_len": None, "num_heads": 1, "scale_value": 1.0, "pre_tokens": 2147483647, "next_tokens": 2147483647, "input_layout": 'BSH', "num_key_value_heads": 0, "sparse_mode": 0, "inner_precise": 1, "block_size": 0, "antiquant_mode": 0, "softmax_lse_flag": False, "key_antiquant_mode": 0, "value_antiquant_mode": 0},
+    "GroupedMatmul": {"bias": None, "scale": None, "offset": None, "antiquant_scale": None, "antiquant_offset": None, "group_list": None, "split_item": 0, "group_type": -1, "transpose_a": False, "transpose_b": False},
     "GroupedMatmulV2": {"bias": None, "scale": None, "offset": None, "antiquant_scale": None, "antiquant_offset": None, "group_list": None, "split_item": 0, "group_type": -1},
     "GroupedMatmulV4": {"bias": None, "scale": None, "offset": None, "antiquant_scale": None, "antiquant_offset": None, "pre_token_scale": None, "group_list": None, "activation_input": None, "activation_quant_scale": None, "activation_quant_offset": None, "split_item": 0, "group_type": -1, "group_list_type": 0, "act_type": 0},
     "KVCacheScatterUpdate": {"reduce": 'none'},
@@ -349,6 +381,8 @@ op_args_default_value = {
     "MoeGatingTopKSoftmax": {"finished": None, "k": 1},
     "QuantBatchMatmul": {"offset": None, "bias": None, "pertokenScaleOptional": None, "transpose_x1": False, "transpose_x2": False, "dtype": mstype.float16},
     "QuantV2": {"sqrt_mode": False, "rounding_mode": 'ROUND', "dst_type": mstype.int8},
+    "RmsNormQuant": {"beta": None, "epsilon": 1e-6},
+    "TransposeBatchMatmulTranspose": {"transpose_a": False, "transpose_b": False},
     "WeightQuantBatchMatmul": {"antiquant_offset": None, "quant_scale": None, "quant_offset": None, "bias": None, "transpose_x": False, "transpose_weight": False, "antiquant_group_size": 0},
     "DeprecatedAddbmm": {"beta": 1, "alpha": 1},
     "DeprecatedAddmm": {"beta": 1, "alpha": 1},
@@ -360,14 +394,19 @@ op_args_default_value = {
     "DeprecatedArgmax": {"axis": -1, "keepdims": False},
     "DeprecatedArgmin": {"axis": None, "keepdims": False},
     "DeprecatedArgsort": {"axis": -1, "descending": False},
+    "DeprecatedBaddbmm": {"beta": 1, "alpha": 1},
     "DeprecatedBincount": {"weights": None, "minlength": 0},
     "DeprecatedChunk": {"axis": 0},
+    "DeprecatedInplaceCopy": {"non_blocking": False},
+    "DeprecatedCountNonzero": {"axis": (), "keep_dims": False, "dtype": None},
     "DeprecatedCumsum": {"axis": None, "dtype": None},
     "DeprecatedDiv": {"rounding_mode": None},
     "DeprecatedFlatten": {"order": 'C', "start_dim": 0, "end_dim": -1},
     "DeprecatedGather": {"batch_dims": 0},
     "DeprecatedHistc": {"bins": 100, "min": 0, "max": 0},
+    "DeprecatedIndexAdd": {"use_lock": True, "check_index_bound": True},
     "DeprecatedIsclose": {"rtol": 1e-05, "atol": 1e-08, "equal_nan": False},
+    "DeprecatedLogsumexp": {"keepdim": False},
     "DeprecatedMax": {"axis": None, "keepdims": False, "initial": None, "where": True, "return_indices": False},
     "DeprecatedMean": {"axis": None, "keep_dims": False},
     "DeprecatedMedian": {"axis": -1, "keepdims": False},
@@ -375,10 +414,12 @@ op_args_default_value = {
     "DeprecatedNansum": {"axis": None, "keepdims": False, "dtype": None},
     "DeprecatedProd": {"dim": None, "keepdim": False, "dtype": None},
     "DeprecatedRepeatInterleave": {"dim": None},
+    "DeprecatedRoll": {"dims": None},
     "DeprecatedSort": {"axis": -1, "descending": False},
     "DeprecatedSplit": {"axis": 0},
     "DeprecatedStd": {"axis": None, "ddof": 0, "keepdims": False},
     "DeprecatedSum": {"axis": None, "dtype": None, "keepdims": False, "initial": None},
+    "DeprecatedTake": {"axis": None, "mode": 'clip'},
     "DeprecatedTopk": {"dim": None, "largest": True, "sorted": True},
     "DeprecatedTranspose": {"axes": None},
     "DeprecatedTril": {"diagonal": 0},
@@ -388,8 +429,8 @@ op_args_default_value = {
 }
 
 op_labels = {
-    "AdamWeightDecay": {"side_effect_mem": True},
     "AdamW": {"side_effect_mem": True},
+    "AdamWeightDecay": {"side_effect_mem": True},
     "ApplyAdamW": {"side_effect_mem": True},
     "AssignAdd": {"side_effect_mem": True},
     "Assign": {"side_effect_mem": True},
@@ -402,36 +443,45 @@ op_labels = {
     "DropoutExt": {"side_effect_hidden": True},
     "DropoutGenMaskExt": {"side_effect_hidden": True},
     "Dropout": {"side_effect_hidden": True},
-    "EmbeddingApplyAdaGrad": {"_process_node_engine_id": 'PS'},
-    "EmbeddingApplyAdam": {"_process_node_engine_id": 'PS'},
     "EmbeddingApplyAdamW": {"_process_node_engine_id": 'PS'},
+    "EmbeddingApplyAdam": {"_process_node_engine_id": 'PS'},
+    "EmbeddingApplyAdaGrad": {"_process_node_engine_id": 'PS'},
     "EmbeddingApplyFtrl": {"_process_node_engine_id": 'PS'},
     "EmbeddingApplyRmsprop": {"_process_node_engine_id": 'PS'},
     "EmbeddingApplySgd": {"_process_node_engine_id": 'PS'},
     "Embedding": {"side_effect_mem": True},
     "EmbeddingTableEvict": {"_process_node_engine_id": 'PS'},
     "Generator": {"side_effect_mem": True},
+    "GroupTopk": {"side_effect_mem": True},
     "InnerInplaceIndexPut": {"side_effect_mem": True},
-    "InplaceAddExt": {"side_effect_mem": True},
     "InplaceAddmm": {"side_effect_mem": True},
     "InplaceAddsExt": {"side_effect_mem": True},
+    "InplaceAddExt": {"side_effect_mem": True},
     "InplaceCopy": {"side_effect_mem": True},
-    "InplaceDiv": {"side_effect_mem": True},
-    "InplaceDivMod": {"side_effect_mem": True},
     "InplaceDivMods": {"side_effect_mem": True},
+    "InplaceDivMod": {"side_effect_mem": True},
     "InplaceDivs": {"side_effect_mem": True},
+    "InplaceDiv": {"side_effect_mem": True},
+    "InplaceElu": {"side_effect_mem": True},
+    "InplaceErfinv": {"side_effect_mem": True},
     "InplaceExp": {"side_effect_mem": True},
+    "InplaceFillDiagonal": {"side_effect_mem": True},
     "InplaceFillScalar": {"side_effect_mem": True},
     "InplaceFillTensor": {"side_effect_mem": True},
+    "InplaceFloorDivides": {"side_effect_mem": True},
+    "InplaceFloorDivide": {"side_effect_mem": True},
     "InplaceFloor": {"side_effect_mem": True},
+    "InplaceGroupedMatmulAdd": {"side_effect_mem": True},
     "InplaceHardtanh": {"side_effect_mem": True},
     "InplaceIndexAddExt": {"side_effect_mem": True},
     "InplaceIndexPut": {"side_effect_mem": True},
+    "InplaceLog": {"side_effect_mem": True},
     "InplaceMaskedFillScalar": {"side_effect_mem": True},
     "InplaceMaskedFillTensor": {"side_effect_mem": True},
-    "InplaceMul": {"side_effect_mem": True},
     "InplaceMuls": {"side_effect_mem": True},
+    "InplaceMul": {"side_effect_mem": True},
     "InplaceNormal": {"side_effect_mem": True},
+    "InplacePut": {"side_effect_mem": True},
     "InplaceRandom": {"side_effect_mem": True},
     "InplaceReLU": {"side_effect_mem": True},
     "InplaceScatterAdd": {"side_effect_mem": True},
@@ -442,6 +492,8 @@ op_labels = {
     "InplaceSubExt": {"side_effect_mem": True},
     "InplaceSubScalar": {"side_effect_mem": True},
     "InplaceTanh": {"side_effect_mem": True},
+    "InplaceThreshold": {"side_effect_mem": True},
+    "InplaceUniform": {"side_effect_mem": True},
     "Log": {"cust_aicpu": 'Log', "base": -1.0, "scale": 1.0, "shift": 0.0},
     "PromptKVCache": {"side_effect_mem": True},
     "ReshapeAndCache": {"side_effect_mem": True},
@@ -449,4 +501,5 @@ op_labels = {
     "SilentCheckV2": {"side_effect_mem": True},
     "SilentCheckV3": {"side_effect_mem": True},
     "KVCacheScatterUpdate": {"side_effect_mem": True},
+    "DeprecatedInplaceCopy": {"side_effect_mem": True},
 }