mindspore 2.5.0__cp310-cp310-win_amd64.whl → 2.6.0rc1__cp310-cp310-win_amd64.whl

mindspore/ops_generate/gen_constants.py

@@ -1,190 +0,0 @@
-# Copyright 2024 Huawei Technologies Co., Ltd
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-# ============================================================================
-"""
-This module contains the constant strings used in generating ops files.
-
-Constants:
-    PY_LICENSE: License strings used for .py files
-    CC_LICENSE: License strings used for .h/.cc files
-    ......
-    Other constant strings in the module are used for generation paths
-"""
-
-import os
-
-# py license
-PY_LICENSE = f"""# Copyright 2024 Huawei Technologies Co., Ltd
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-# ============================================================================
-"""
-
-# cc license
-CC_LICENSE = f"""/**
- * Copyright 2024 Huawei Technologies Co., Ltd
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */"""
-
-# op_def
-OP_DEF_AUTO_GENERATE_PATH = "op_def/auto_generate"
-MS_OP_DEF_AUTO_GENERATE_PATH = "mindspore/ops/op_def/auto_generate"
-MS_OP_DEF_YAML_PATH = "mindspore/ops/op_def/yaml"
-MS_OP_DEPRECATED_DEF_YAML_PATH = "mindspore/ops/op_def/deprecated"
-MS_OP_API_YAML_PATH = "mindspore/ops/api_def"
-MS_TENSOR_METHOD_DOC_YAML_PATH = "mindspore/ops/api_def/method_doc"
-MS_MINT_FUNC_DOC_YAML_PATH = "mindspore/ops/api_def/function_doc"
-MS_MINT_FUNC_OVERLOAD_PATH = "mindspore/python/mindspore/ops"
-PY_AUTO_GEN_PATH = "mindspore/python/mindspore/ops/auto_generate"
-PY_OPS_GEN_PATH = "mindspore/python/mindspore/ops_generate"
-PYBOOST_NATIVE_GRAD_FUNC_GEN_PATH = "mindspore/ccsrc/pipeline/pynative/grad/function/auto_generate"
-PYBOOST_AUTO_GRAD_FUNC_GEN_PATH = "mindspore/ccsrc/pipeline/pynative/op_function/auto_generate"
-PIPELINE_PYBOOST_FUNC_GEN_PATH = "mindspore/ccsrc/pipeline/pynative/op_function/auto_generate"
-RUNTIME_PYBOOST_FUNC_GEN_PATH = "mindspore/ccsrc/runtime/pynative/op_function/auto_generate"
-TENSOR_FUNC_REGISTER_PATH = "mindspore/ccsrc/pybind_api/ir/tensor_register/auto_generate"
-TENSOR_API_PATH = "mindspore/ccsrc/pybind_api/ir/tensor_api/auto_generate"
-ADD_TENSOR_DOCS_PY_PATH = "mindspore/python/mindspore/common"
-ADD_MINT_DOCS_PY_PATH = "mindspore/python/mindspore/mint"
-
-# yaml keys def
-OP_KEYS = {'args', 'args_signature', 'returns', 'function', 'class', 'view', 'dispatch', 'labels', 'bprop_expander'}
-ARG_KEYS = {'dtype', 'default', 'prim_init', 'type_cast', 'arg_handler'}
-RETURN_KEYS = {'dtype', 'inplace', 'type_cast'}
-ARG_SIGNATURE_KEYS = {'rw_write', 'rw_read', 'rw_ref', 'dtype_group'}
-CLASS_KEYS = {'name', 'disable'}
-FUNCTION_KEYS = {'name', 'disable'}
-DISPATCH_KEYS = {'enable', 'is_comm_op', 'Ascend', 'GPU', 'CPU'}
-TENSOR_FUNC_KEYS = {'op_yaml', 'py_method', 'kwonlyargs', 'varargs', 'alias', 'Ascend', 'GPU', 'CPU', 'interface'}
-
-# func signature parsing
-ARG_HANDLER_MAP = {"to_2d_paddings": "int|tuple[int]|list[int]",
-                   "dtype_to_type_id": "type",
-                   "to_kernel_size": "int|tuple[int]|list[int]",
-                   "to_strides": "int|tuple[int]|list[int]",
-                   "str_to_enum": "str",
-                   "to_pair": "int|tuple[int]|list[int]|float",
-                   "to_dilations": "tuple[int]|list[int]|int",
-                   "to_output_padding": "int|tuple[int]|list[int]",
-                   "to_rates": "int|tuple[int]|list[int]"}
-INPUT_ARGS_NAME = {"input", "x", "input_x"}
-INPUT_NAME_MAP = {"DeprecatedExpandAs": "input"}
-
-# infer
-MS_OPS_FUNC_IMPL_PATH = "mindspore/ops/infer/ops_func_impl"
-
-# view
-MS_OPS_VIEW_PATH = "mindspore/ops/view"
-
-# kernel
-MS_OPS_KERNEL_PATH = "mindspore/ops/kernel"
-MS_OPS_KERNEL_FUNCTIONS_AUTO_GEN_PATH = "mindspore/ops/kernel/functions/auto_generate"
-MS_COMMON_PYBOOST_KERNEL_PATH = os.path.join(MS_OPS_KERNEL_PATH, "common/pyboost")
-
-
-OP_NAME_OP_DEF = """
-#ifndef MINDSPORE_CORE_OP_NAME_H_
-#define MINDSPORE_CORE_OP_NAME_H_
-
-namespace mindspore::ops {
-$ops_namespace_body
-}  // namespace mindspore::ops
-
-#endif  // MINDSPORE_CORE_OP_NAME_H_
-"""
-
-OP_PRIM_OP_DEF = """
-#ifndef MINDSPORE_CORE_OPS_GEN_OPS_PRIMITIVE_H_
-#define MINDSPORE_CORE_OPS_GEN_OPS_PRIMITIVE_H_
-
-#include <memory>
-#include "ir/anf.h"
-#include "ir/primitive.h"
-#include "$auto_gen_path/gen_ops_name.h"
-#include "mindapi/base/macros.h"
-
-namespace mindspore::prim {
-$ops_prim_gen
-}  // namespace mindspore::prim
-#endif  // MINDSPORE_CORE_OPS_GEN_OPS_PRIMITIVE_H_
-"""
-
-LITE_OPS_CC = """
-#include "$auto_gen_path/gen_lite_ops.h"
-#include "mindapi/helper.h"
-#include "ops/primitive_c.h"
-#include "ops/base_operator.h"
-#include "abstract/abstract_value.h"
-
-namespace mindspore::ops {
-$ops_namespace_body
-
-}  // namespace mindspore::ops
-    """
-
-LITE_OPS_H = """
-#ifndef MINDSPORE_CORE_OPS_GEN_LITE_OPS_H_
-#define MINDSPORE_CORE_OPS_GEN_LITE_OPS_H_
-
-#include <vector>
-#include "ops/base_operator.h"
-#include "$auto_gen_path/gen_ops_name.h"
-
-namespace mindspore::ops {
-$ops_namespace_body
-
-}  // namespace mindspore::ops
-#endif  // MINDSPORE_CORE_OPS_GEN_LITE_OPS_H_
-"""
-
-CC_OPS_DEF = """
-
-#include "$auto_generate_path/gen_ops_def.h"
-#include "ir/signature.h"
-$gen_include
-
-namespace mindspore::ops {$gen_cc_code
-}  // namespace mindspore::ops
-"""
-
-ACLNN_REG_CODE = """
-#include "$ops_gen_kernel_path/ascend/opapi/aclnn_kernel_mod.h"
-
-namespace mindspore {
-namespace kernel {
-
-$aclnn_reg_code
-}  // namespace kernel
-}  // namespace mindspore
-"""

mindspore/ops_generate/gen_ops_inner_prim.py

@@ -1,131 +0,0 @@
-# Copyright 2023-2024 Huawei Technologies Co., Ltd
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-# ============================================================================
-"""Primitive defined for arg handler."""
-
-from mindspore.ops.primitive import Primitive, prim_attr_register, prim_arg_register
-from mindspore._c_expression import typing
-from mindspore._c_expression import op_enum
-
-
-class DtypeToEnum(Primitive):
-    r"""
-    Convert mindspore dtype to enum.
-
-    Inputs:
-        - **op_name** (str) - The op name
-        - **arg_name** (str) - The arg name
-        - **dtype** (mindspore.dtype) - The data type.
-
-    Outputs:
-        An integer.
-
-    Supported Platforms:
-        ``Ascend`` ``GPU`` ``CPU``
-    """
-
-    @prim_attr_register
-    def __init__(self):
-        """Initialize"""
-
-    def __call__(self, op_name, arg_name, dtype):
-        """Run in PyNative mode"""
-        if not isinstance(dtype, typing.Type):
-            raise TypeError(f"For '{op_name}', the input '{arg_name}' should be mindspore dtype, but got {dtype}.")
-        return typing.type_to_type_id(dtype)
-
-
-class StringToEnum(Primitive):
-    r"""
-    Convert string to enum.
-
-    Inputs:
-        - **op_name** (str) - The op name
-        - **arg_name** (str) - The arg name
-        - **enum_str** (str) - The str data.
-
-    Outputs:
-        An integer.
-
-    Supported Platforms:
-        ``CPU``
-    """
-
-    @prim_attr_register
-    def __init__(self):
-        """Initialize"""
-
-    def __call__(self, op_name, arg_name, enum_str):
-        """Run in PyNative mode"""
-        if enum_str is None:
-            return None
-        if not isinstance(enum_str, str):
-            raise TypeError(f"For '{op_name}', the input '{arg_name}' should be a str, but got {type(enum_str)}.")
-        return op_enum.str_to_enum(op_name, arg_name, enum_str)
-
-
-class TupleToList(Primitive):
-    r"""
-    Convert tuple to list.
-
-    Inputs:
-        - **x** (tuple) - The input
-
-    Outputs:
-        List, has the same elements as the `input`.
-
-    Supported Platforms:
-        ``CPU``
-
-    Examples:
-        >>> from mindspore.ops_generate import TupleToList
-        >>> x = (1, 2, 3)
-        >>> result = TupleToList()(x)
-        >>> print(result)
-        [1, 2, 3]
-    """
-    @prim_arg_register
-    def __init__(self):
-        """Initialize TupleToList"""
-
-    def __call__(self, input):
-        return list(input)
-
-
-class ListToTuple(Primitive):
-    r"""
-    Convert list to tuple.
-
-    Inputs:
-        - **x** (list) - The input
-
-    Outputs:
-        Tuple, has the same elements as the `input`.
-
-    Supported Platforms:
-        ``CPU``
-
-    Examples:
-        >>> from mindspore.ops_generate import ListToTuple
-        >>> x = [1, 2, 3]
-        >>> result = ListToTuple()(x)
-        >>> print(result)
-        (1, 2, 3)
-    """
-    @prim_arg_register
-    def __init__(self):
-        """Initialize TupleToList"""
-
-    def __call__(self, input):
-        return tuple(input)

mindspore/ops_generate/ops_primitive_h_generator.py

@@ -1,81 +0,0 @@
-# Copyright 2024 Huawei Technologies Co., Ltd
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-# ============================================================================
-"""
-Module for generating C++ header files for operator primitives.
-
-This module defines the `OpsPrimitiveHGenerator` class, which creates C++ header files
-containing definitions for operator primitives based on provided operator prototypes.
-"""
-
-import os
-
-import gen_constants as K
-import gen_utils
-import pyboost_utils
-
-# refactored
-import template
-
-from base_generator import BaseGenerator
-
-
-class OpsPrimitiveHGenerator(BaseGenerator):
-    """
-    This class generates the header file for operator primitives.
-    """
-
-    def __init__(self):
-        """
-        Initializes the generator with templates for operator primitive definitions.
-        """
-        self.op_prim_op_def_template = template.Template(K.OP_PRIM_OP_DEF)
-        self.op_def_template = template.Template(
-            "GVAR_DEF(PrimitivePtr, kPrim${k_name_op}, std::make_shared<Primitive>(ops::kName${k_name_op}))\n")
-        self.op_def_rw_template = template.Template(
-            "GVAR_DEF(PrimitivePtr, kPrim${k_name_op}, std::make_shared<Primitive>(ops::kName${k_name_op}, "
-            "true, kPrimTypeBuiltIn, true))\n")
-
-    def generate(self, work_path, op_protos):
-        """
-        Generates the header file content for operator primitives and saves it.
-
-        Args:
-            work_path (str): The directory to save the generated files.
-            op_protos (list): A list of operator prototypes.
-
-        Returns:
-            None
-
-        The method generates the content of the header file for each operator primitive
-        defined in the 'op_protos' list and saves it to the specified work path.
-        """
-        ops_prim_gen_list = []
-        for op_proto in op_protos:
-            k_name_op = pyboost_utils.get_op_name(op_proto.op_name, op_proto.op_class.name)
-            if op_proto.op_args_signature:
-                if op_proto.op_args_signature.rw_write:
-                    ops_prim_gen_list.append(self.op_def_rw_template.replace(k_name_op=k_name_op))
-                    continue
-
-            ops_prim_gen_list.append(self.op_def_template.replace(k_name_op=k_name_op))
-
-        op_prim_op_def = self.op_prim_op_def_template.replace(auto_gen_path=K.MS_OP_DEF_AUTO_GENERATE_PATH,
-                                                              ops_prim_gen=ops_prim_gen_list)
-
-        res_str = template.CC_LICENSE_STR + op_prim_op_def
-
-        save_path = os.path.join(work_path, K.MS_OP_DEF_AUTO_GENERATE_PATH)
-        file_name = "gen_ops_primitive.h"
-        gen_utils.save_file(save_path, file_name, res_str)