mindspore 2.3.0__cp39-none-any.whl → 2.3.0rc2__cp39-none-any.whl

mindspore/ops/auto_generate/cpp_create_prim_instance_helper.py

@@ -17,15 +17,13 @@
 from mindspore.common import dtype as mstype
 
 op_args_default_value = {
+    "AdamWeightDecayExt": {"amsgrad": False, "maximize": False},
     "AdamWeightDecay": {"use_locking": False},
-    "AdamW": {"amsgrad": False, "maximize": False},
     "AddExt": {"alpha": 1},
-    "AddLayerNormV2": {"epsilon": 1e-5, "additionalOut": False},
     "ApplyCamePart2": {"sum_r": None, "global_shape": None},
     "ApplyCamePart3": {"global_shape": None, "use_first_moment": False},
     "ApplyCamePart4": {"global_shape": None},
     "ApplyRotaryPosEmb": {"cos_format": 0},
-    "Arange": {"dtype": None},
     "ArgMaxExt": {"dim": None, "keepdim": False},
     "Argmax": {"axis": -1, "output_type": mstype.int32},
     "ArgMaxWithValue": {"axis": 0, "keep_dims": False},
@@ -46,18 +44,12 @@ op_args_default_value = {
     "BatchNormWithAddAndActivation": {"is_training": False, "epsilon": 1e-5, "momentum": 0.1, "data_format": 'NCHW'},
     "BiasAddGrad": {"data_format": 'NCHW'},
     "BiasAdd": {"data_format": 'NCHW'},
-    "BinaryCrossEntropyGrad": {"weight": None, "reduction": 'mean'},
-    "BinaryCrossEntropy": {"weight": None, "reduction": 'mean'},
-    "BinaryCrossEntropyWithLogitsBackward": {"weight": None, "posWeight": None, "reduction": 'mean'},
-    "BCEWithLogitsLoss": {"weight": None, "posWeight": None, "reduction": 'mean'},
     "CeLU": {"alpha": 1.0},
     "CholeskyInverse": {"upper": False},
     "Cholesky": {"upper": False},
     "Chunk": {"dim": 0},
     "ClampScalar": {"min": None, "max": None},
     "ClampTensor": {"min": None, "max": None},
-    "Col2ImExt": {"dilation": 1, "padding": 0, "stride": 1},
-    "Col2ImGrad": {"dilation": 1, "padding": 0, "stride": 1},
     "Concat": {"axis": 0},
     "ConstantPadND": {"value": 0.0},
     "ConvolutionGrad": {"bias": None, "stride": 1, "padding": 0, "dilation": 1, "transposed": False, "output_padding": 0, "groups": 1, "output_mask": ()},
@@ -65,15 +57,13 @@ op_args_default_value = {
     "Correlate": {"mode": 'valid'},
     "CumProd": {"exclusive": False, "reverse": False},
     "CumSum": {"exclusive": False, "reverse": False},
-    "CumsumExt": {"dtype": None},
     "DCT": {"axis": -1, "norm": 'BACKWARD', "forward": True, "grad": False},
     "Dense": {"bias": None},
     "Diagonal": {"offset": 0, "dim1": 0, "dim2": 1},
     "DivMod": {"rounding_mode": None},
+    "DropoutExt": {"p": 0.5, "seed": 0, "offset": 0},
     "Dropout": {"keep_prob": 0.5, "Seed0": 0, "Seed1": 0},
     "Eig": {"compute_v": False},
-    "EluExt": {"alpha": 1.0},
-    "EluGradExt": {"alpha": 1.0},
     "Elu": {"alpha": 1.0},
     "EmbeddingDenseBackward": {"padding_idx": None, "scale_grad_by_freq": False},
     "Embedding": {"padding_idx": None, "max_norm": None, "norm_type": 2.0, "scale_grad_by_freq": False},
@@ -102,11 +92,8 @@ op_args_default_value = {
     "IFFT": {"n": None, "dim": -1, "norm": None},
     "IFFTN": {"s": None, "dim": None, "norm": None},
     "IFFTShift": {"dim": None},
-    "Im2ColExt": {"dilation": 1, "padding": 0, "stride": 1},
-    "IndexAddExt": {"alpha": 1},
     "IRFFTGrad": {"n": None, "dim": -1, "norm": None},
     "IRFFT": {"n": None, "dim": -1, "norm": None},
-    "IsClose": {"rtol": 1e-05, "atol": 1e-08, "equal_nan": True},
     "LayerNormExt": {"weight": None, "bias": None, "eps": 1e-5},
     "LayerNormGradGrad": {"begin_norm_axis": 1, "begin_params_axis": 1},
     "LayerNormGrad": {"begin_norm_axis": 1, "begin_params_axis": 1},
@@ -129,6 +116,7 @@ op_args_default_value = {
     "MaximumGrad": {"grad_x": True, "grad_y": True},
     "MeanExt": {"axis": None, "keep_dims": False, "dtype": None},
     "MinimumGrad": {"grad_x": True, "grad_y": True},
+    "MoeFinalizeRouting": {"x2": None, "bias": None, "scales": None, "expanded_row_idx": None, "expanded_expert_idx": None},
     "NanToNum": {"nan": None, "posinf": None, "neginf": None},
     "NLLLossGrad": {"reduction": 'mean', "ignore_index": -100},
     "NLLLoss": {"reduction": 'mean', "ignore_index": -100},
@@ -140,8 +128,7 @@ op_args_default_value = {
     "ProdExt": {"axis": None, "keep_dims": False, "dtype": None},
     "PromptKVCache": {"align_mode": 'LEFT'},
     "Qr": {"full_matrices": False},
-    "RandExt": {"dtype": None},
-    "RandLikeExt": {"dtype": None},
+    "QuantBatchMatmul": {"offset": None, "bias": None, "transpose_x1": False, "transpose_x2": False, "dtype": mstype.float16},
     "RandpermV2": {"seed": 0, "offset": 0, "dtype": mstype.int64},
     "Range": {"maxlen": 1000000},
     "ReduceAll": {"axis": None, "keep_dims": False},
@@ -152,8 +139,7 @@ op_args_default_value = {
     "ReduceProd": {"axis": (), "keep_dims": False},
     "ReduceStd": {"axis": [], "unbiased": True, "keep_dims": False},
     "ReduceSum": {"axis": (), "keep_dims": False, "skip_mode": False},
-    "RepeatInterleaveInt": {"dim": None, "output_size": None},
-    "RepeatInterleaveTensor": {"dim": None, "output_size": None},
+    "RepeatInterleave": {"axis": None, "output_size": 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},
@@ -167,16 +153,13 @@ op_args_default_value = {
     "ResizeNearestNeighborV2": {"align_corners": False, "half_pixel_centers": False},
     "RFFTGrad": {"n": None, "dim": -1, "norm": None},
     "RFFT": {"n": None, "dim": -1, "norm": None},
-    "RmsNorm": {"epsilon": 1e-6},
     "ScalarToTensor": {"dtype": None},
-    "SearchSorted": {"sorter": None, "dtype": mstype.int64, "right": False},
     "SequenceConcat": {"axis": 0},
     "SoftmaxBackward": {"dim": -1},
     "Softmax": {"axis": -1},
     "SoftplusExt": {"beta": 1, "threshold": 20},
     "SoftplusGradExt": {"beta": 1, "threshold": 20},
     "SolveTriangular": {"trans": 0, "lower": False, "unit_diagonal": False},
-    "SortExt": {"dim": -1, "descending": False, "stable": False},
     "Split": {"axis": 0, "output_num": 1},
     "SplitTensor": {"axis": 0},
     "SplitWithSize": {"axis": 0},
@@ -185,10 +168,9 @@ op_args_default_value = {
     "SubExt": {"alpha": 1},
     "SumExt": {"dim": None, "keepdim": False, "dtype": None},
     "TopkExt": {"dim": -1, "largest": True, "sorted": True},
+    "Tril": {"diagonal": 0},
     "Triu": {"diagonal": 0},
     "TupleToTensor": {"dtype": None},
-    "Unique2": {"sorted": True, "return_inverse": False, "return_counts": False},
-    "UnstackExt": {"axis": 0},
     "UpsampleBilinear2DGrad": {"output_size": None, "scales": None, "align_corners": False},
     "UpsampleBilinear2D": {"output_size": None, "scales": None, "align_corners": False},
     "UpsampleLinear1DGrad": {"output_size": None, "scales": None, "align_corners": False},
@@ -201,21 +183,14 @@ op_args_default_value = {
     "UpsampleNearest3D": {"output_size": None, "scales": None},
     "UpsampleTrilinear3DGrad": {"output_size": None, "scales": None, "align_corners": False},
     "UpsampleTrilinear3D": {"output_size": None, "scales": None, "align_corners": False},
+    "WeightQuantBatchMatmul": {"antiquant_offset": None, "quant_scale": None, "quant_offset": None, "bias": None, "transpose_x": False, "transpose_weight": False, "antiquant_group_size": 0},
     "ZerosLikeExt": {"dtype": None},
     "Zeros": {"dtype": None},
-    "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},
-    "KVCacheScatterUpdate": {"reduce": 'none'},
-    "MoeFinalizeRouting": {"x2": None, "bias": None, "scales": None, "expanded_row_idx": None, "expanded_expert_idx": None},
-    "QuantBatchMatmul": {"offset": None, "bias": None, "transpose_x1": False, "transpose_x2": False, "dtype": mstype.float16},
-    "QuantV2": {"sqrt_mode": False, "rounding_mode": 'ROUND', "dst_type": mstype.int8},
-    "WeightQuantBatchMatmul": {"antiquant_offset": None, "quant_scale": None, "quant_offset": None, "bias": None, "transpose_x": False, "transpose_weight": False, "antiquant_group_size": 0},
 }
 
 op_labels = {
+    "AdamWeightDecayExt": {"side_effect_mem": True},
     "AdamWeightDecay": {"side_effect_mem": True},
-    "AdamW": {"side_effect_mem": True},
     "AssignAdd": {"side_effect_mem": True},
     "Assign": {"side_effect_mem": True},
     "DecoderKVCache": {"side_effect_mem": True},
@@ -223,7 +198,6 @@ op_labels = {
     "DropoutGenMaskExt": {"side_effect_hidden": True},
     "Dropout": {"side_effect_hidden": True},
     "Embedding": {"side_effect_mem": True},
-    "Generator": {"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},

mindspore/ops/auto_generate/gen_arg_dtype_cast.py

@@ -20,18 +20,25 @@ import mindspore as ms
 from mindspore import ops
 from mindspore.common.tensor import Tensor
 from mindspore.ops.operations._sequence_ops import TensorToScalar, TensorToTuple
-from mindspore.ops_generate.gen_ops_inner_prim import TupleToList, ListToTuple
+from mindspore.ops_generate.gen_ops_inner_prim import TupleToList
 from mindspore._c_expression import OpDtype
 
 tensor_to_tuple_ = TensorToTuple()
 tuple_to_list = TupleToList()
-list_to_tuple = ListToTuple()
 
 
 def int_to_float(data):
     return float(data)
 
 
+def list_to_tuple(data):
+    # tuple() currently does not support Any from JIT Fallback.
+    res = ()
+    for element in data:
+        res += (element,)
+    return res
+
+
 def scalar_to_tuple(data):
     return (data,)
 

mindspore/ops/auto_generate/gen_arg_handler.py

@@ -18,8 +18,6 @@ from mindspore.ops_generate.gen_ops_inner_prim import DtypeToEnum, StringToEnum
 # Enum Class:
 from mindspore._c_expression import FormatEnum as Format
 from mindspore._c_expression import ReductionEnum as Reduction
-from mindspore.common import Tensor
-from mindspore.common import dtype as mstype
 
 
 def arg_invalid_info(op_name, arg_name, arg_val):
@@ -29,17 +27,6 @@ def arg_invalid_info(op_name, arg_name, arg_val):
     return f"For '{op_name}', the value of '{arg_name}' is invalid: '{arg_val}'."
 
 
-def to_pair(op_name, arg_name, arg_val):
-    """
-    convert arg_val: int/tuple[int*2] -> tuple[int*2].
-    """
-    if isinstance(arg_val, (int, float)):
-        return (arg_val, arg_val)
-    if isinstance(arg_val, (list, tuple)):
-        return arg_val
-    raise ValueError(arg_invalid_info(op_name, arg_name, arg_val))
-
-
 def to_kernel_size(op_name, arg_name, kernel_size):
     """
     convert kernel_size: int/tuple[int*4] -> tuple[int*2].
@@ -176,19 +163,6 @@ def to_3d_paddings(op_name, arg_name, pad):
         return pad
     raise ValueError(arg_invalid_info(op_name, arg_name, pad))
 
-
-def generator_handler(op_name, arg_name, inputs):
-    """
-    convert constant value in tuple to tensor
-    """
-    new_inputs = []
-    for input_ in inputs:
-        if isinstance(input_, int):
-            new_inputs.append(Tensor(input_, mstype.int64))
-        else:
-            new_inputs.append(input_)
-    return tuple(new_inputs)
-
 dtype_to_type_id = DtypeToEnum()
 
 # string to enum