mindspore 2.7.0rc1__cp311-cp311-win_amd64.whl → 2.7.1__cp311-cp311-win_amd64.whl

mindspore/ops_generate/common/template.py

@@ -13,94 +13,27 @@
 # limitations under the License.
 # ============================================================================
 """Templates for code auto generation."""
-import re
+__all__ = [
+    "Template",
+    "NEW_LINE",
+    "PYTHON_PRIM_TEMPLATE",
+    "IMPORT_PYBOOST_PRIM_HEADER",
+    "IMPORT_PYBOOST_FUNC_HEADER",
+    "PY_LICENSE_STR",
+    "OPS_PY_PRIM_HEADER",
+    "OPS_PY_DEF_HEADER",
+    "CUSTOM_OPS_PY_DEF_HEADER",
+    "PRIMITIVE_CLASS_DESC",
+    "CC_LICENSE_STR",
+    "OP_PROTO_TEMPLATE",
+    "OP_PRIM_CLASS_DEFINE_TEMPLATE",
+]
+
 import os
 import common.gen_constants as K
-
-
-class Template:
-    """
-    template for generate c++/python code
-    """
-    regular_str = r"(^[^\n\S]*)?\$([^\d\W]\w*|\{,?[^\d\W]\w*\,?})"
-    regular_match = re.compile(regular_str, re.MULTILINE)
-
-    def __init__(self, code_pattern):
-        self.code_pattern = code_pattern
-
-    @staticmethod
-    def load_from_file(file_path):
-        with open(file_path, "r") as f:
-            return Template(f.read())
-
-    def replace(self, **kwargs):
-        """
-        replace param.
-        :param kwargs:
-        :return:
-        """
-
-        def find(key: str):
-            if key in kwargs:
-                return kwargs[key]
-            raise TypeError(f"{key} should be in kwargs!")
-
-        def add_indent(indent, var):
-            return "".join([indent + line + "\n" for data in var for line in str(data).splitlines()]).rstrip()
-
-        def extract_variable(key):
-            start = ""
-            end = ""
-            if key[0] == "{":
-                key = key[1:-1]
-                if key[0] == ",":
-                    start = ","
-                    key = key[1:]
-                if key[-1] == ",":
-                    end = ", "
-                    key = key[:-1]
-            return find(key), start, end
-
-        def match_rule(match):
-            indent = match.group(1)
-            key = match.group(2)
-            var, start, end = extract_variable(key)
-            if indent is not None:
-                if not isinstance(var, list):
-                    return add_indent(indent, [var])
-                return add_indent(indent, var)
-            if isinstance(var, list):
-                code = ", ".join(str(x) for x in var)
-                if not var:
-                    return code
-                return start + code + end
-            return str(var)
-
-        return self.regular_match.sub(match_rule, self.code_pattern)
-
-
-NEW_LINE = "\n"
-
-PYTHON_PRIM_TEMPLATE = Template("""
-
-class _Pyboost${class_name}Prim(${class_name}Prim_):
-    def __call__(self, ${input_args}):
-        ${process_func}
-        return super().__call__([${processed_args}])
-
-
-${func_impl_name}_impl = _Pyboost${class_name}Prim()
-""")
-
-IMPORT_PYBOOST_PRIM_HEADER = f"""
-from mindspore.ops._utils.arg_handler import *
-"""
-
-IMPORT_PYBOOST_FUNC_HEADER = f"""
-from mindspore.common import dtype as mstype
-from mindspore.ops.auto_generate.pyboost_inner_prim import *
-
-"""
+from common.template_utils import Template, NEW_LINE, PYTHON_PRIM_TEMPLATE, IMPORT_PYBOOST_PRIM_HEADER, \
+    IMPORT_PYBOOST_FUNC_HEADER, PY_LICENSE_STR, OPS_PY_PRIM_HEADER, OPS_PY_DEF_HEADER, CUSTOM_OPS_PY_DEF_HEADER, \
+    PRIMITIVE_CLASS_DESC, CC_LICENSE_STR, OP_PROTO_TEMPLATE, OP_PRIM_CLASS_DEFINE_TEMPLATE
 
 REGISTER_DEFINE_TEMPLATE = Template(
     """
@@ -117,13 +50,13 @@ REGISTER_PYBOOST_GRAD_DEFINE_TEMPLATE = Template(
 REGISTER_PYBOOST_GRAD_TEMPLATE = Template("${register_func}")
 
 PYBOOST_FUNCTION_HEADER_TEMPLATE = Template.load_from_file(
-    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/op_function/template/pyboost_api_h.tpl'))
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/pyboost_api_h.tpl'))
 
 PYBOOST_CORE_HEADER_TEMPLATE = Template.load_from_file(
-    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/op_function/template/pyboost_core_header.tpl'))
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/pyboost_core_header.tpl'))
 PYBOOST_INTERNAL_OP_HEADER_TEMPLATE = Template.load_from_file(
-    os.path.join(K.WORK_DIR,
-                 f'./{K.MS_OPS_KERNEL_PATH}/ascend/pyboost/internal/template/pyboost_internal_header_template.tpl'))
+    os.path.join(K.WORK_DIR, K.MS_OPS_KERNEL_PATH,
+                 f'ascend/aclnn/pyboost_impl/internal/template/pyboost_internal_header_template.tpl'))
 
 PYBOOST_INTERNAL_SINGLE_OP_HEADER_TEMPLATE = Template.load_from_file(
     os.path.join(K.WORK_DIR,
@@ -178,96 +111,112 @@ PYBOOST_INTERNAL_KERNEL_INFO_ADAPTER_SINGLE_CPP_TEMPLATE = Template.load_from_fi
                  f'{K.MS_PLUGIN_INTERNAL_PATH}/pyboost/template/internal_kernel_info_adapter_single_cpp.tpl'))
 
 PYBOOST_REGISTRY_BODY_CC_TEMPLATE = Template.load_from_file(
-    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/op_function/template/pyboost_registry_body_cc.tpl'))
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/pyboost_registry_body_cc.tpl'))
 
 PYBOOST_CORE_BODY_TEMPLATE = Template.load_from_file(
-    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/op_function/template/pyboost_core_body.tpl'))
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/pyboost_core_body.tpl'))
+
+PYBOOST_CORE_BODY_VIEW_TEMPLATE = Template.load_from_file(
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/pyboost_core_body_view.tpl'))
 
 PYBOOST_CORE_BODY_COMM_TEMPLATE = Template.load_from_file(
-    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/op_function/template/pyboost_core_body_comm.tpl'))
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/pyboost_core_body_comm.tpl'))
 
 PYBOOST_CORE_BODY_SYNC_TEMPLATE = Template.load_from_file(
-    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/op_function/template/pyboost_core_body_sync.tpl'))
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/pyboost_core_body_sync.tpl'))
+
+PYBOOST_CORE_BODY_VIEW_SYNC_TEMPLATE = Template.load_from_file(
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/pyboost_core_body_view_sync.tpl'))
 
 PYBOOST_REGISTRY_CC_TEMPLATE = Template.load_from_file(
-    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/op_function/template/pyboost_registry_cc.tpl'))
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/pyboost_registry_cc.tpl'))
 
 PYBOOST_API_CC_TEMPLATE = Template.load_from_file(
-    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/op_function/template/pyboost_api_cc.tpl'))
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/pyboost_api_cc.tpl'))
 
 PYBOOST_API_BODY_CC_TEMPLATE = Template.load_from_file(
-    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/op_function/template/pyboost_api_body_cc.tpl'))
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/pyboost_api_body_cc.tpl'))
 
 PYBOOST_CORE_CC_TEMPLATE = Template.load_from_file(
-    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/op_function/template/pyboost_core_cc.tpl'))
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/pyboost_core_cc.tpl'))
 
 PYBOOST_OVERLOAD_FUNCTIONS_CC_TEMPLATE = Template.load_from_file(
     os.path.join(
-        K.WORK_DIR, './mindspore/ccsrc/pynative/op_function/template/pyboost_overload_functions_cc.tpl'))
+        K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/pyboost_overload_functions_cc.tpl'))
 
 PYBOOST_MINT_CLASS_DEF = Template.load_from_file(
-    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/op_function/template/pyboost_mint_class_def.tpl'))
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/pyboost_mint_class_def.tpl'))
 
 PYBOOST_OVERLOAD_MINT_CLASS_DEF = Template.load_from_file(
     os.path.join(
-        K.WORK_DIR, './mindspore/ccsrc/pynative/op_function/template/pyboost_overload_mint_class_def.tpl'))
+        K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/pyboost_overload_mint_class_def.tpl'))
 
 # template path need to be moved
 FUNCTIONAL_OVERLOAD_PY_TEMPLATE = Template.load_from_file(
     os.path.join(
-        K.WORK_DIR, './mindspore/ccsrc/pynative/op_function/template/functional_overload_py.tpl')
+        K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/functional_overload_py.tpl')
 )
 
 TENSOR_FUNC_CC_REG = Template.load_from_file(
-    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/op_function/template/tensor_func_cc_reg.tpl'))
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/tensor_func_cc_reg.tpl'))
 
 TENSOR_API_HEADER = Template.load_from_file(
-    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/op_function/template/tensor_api_header.tpl'))
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/tensor_api_header.tpl'))
 
 TENSOR_API_SOURCE = Template.load_from_file(
-    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/op_function/template/tensor_api_source.tpl'))
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/tensor_api_source.tpl'))
 
 TENSOR_FUNC_UTILS = Template.load_from_file(
-    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/op_function/template/tensor_func_utils_header.tpl'))
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/tensor_func_utils_header.tpl'))
 
 TENSOR_FUNC_CALL_BODY = Template.load_from_file(
-    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/op_function/template/tensor_func_call_body.tpl'))
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/tensor_func_call_body.tpl'))
 
 TENSOR_FUNC_OVERLOAD_CALL_BODY = Template.load_from_file(
     os.path.join(K.WORK_DIR,
-                 './mindspore/ccsrc/pynative/op_function/template/tensor_func_overload_call_body.tpl'))
+                 './mindspore/ccsrc/pynative/forward/pyboost/template/tensor_func_overload_call_body.tpl'))
 
 TENSOR_FUNC_UT_BODY = Template.load_from_file(
     os.path.join(K.WORK_DIR,
-                 './mindspore/ccsrc/pynative/op_function/template/tensor_func_ut_body.tpl'))
+                 './mindspore/ccsrc/pynative/forward/pyboost/template/tensor_func_ut_body.tpl'))
 
 TENSOR_FUNC_UT_OVERLOAD_BODY = Template.load_from_file(
     os.path.join(K.WORK_DIR,
-                 './mindspore/ccsrc/pynative/op_function/template/tensor_func_ut_overload_body.tpl'))
+                 './mindspore/ccsrc/pynative/forward/pyboost/template/tensor_func_ut_overload_body.tpl'))
+
+PYBOOST_OVERLOAD_UT_BODY = Template.load_from_file(
+    os.path.join(K.WORK_DIR,
+                 './mindspore/ccsrc/pynative/forward/pyboost/template/pyboost_overload_ut_body.tpl'))
 
 TENSOR_CPP_METHOD = Template.load_from_file(
     os.path.join(K.WORK_DIR,
-                 './mindspore/ccsrc/pynative/op_function/template/tensor_cpp_method.tpl'))
+                 './mindspore/ccsrc/pynative/forward/pyboost/template/tensor_cpp_method.tpl'))
 
 TENSOR_FUNC_CLASS_REG = Template.load_from_file(
-    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/op_function/template/tensor_func_class_reg.tpl'))
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/tensor_func_class_reg.tpl'))
 
 PYBOOST_GRAD_FUNCTION_TEMPLATE = Template.load_from_file(
     os.path.join(K.WORK_DIR, './mindspore/ccsrc/pyboost/grad_functions/template/pyboost_grad_function.tpl'))
 
+PYBOOST_VIEW_GRAD_FUNCTION_TEMPLATE = Template.load_from_file(
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pyboost/grad_functions/template/pyboost_view_grad_function.tpl'))
+
 PYBOOST_GRAD_HEADER_TEMPLATE = Template.load_from_file(
     os.path.join(K.WORK_DIR, './mindspore/ccsrc/pyboost/grad_functions/template/pyboost_grad_function_header.tpl'))
 
 PYBOOST_NATIVE_GRAD_FUNCTION_TEMPLATE = Template.load_from_file(
-    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/grad/function/template/native_grad_function.tpl'))
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/backward/op_grad/template/native_grad_function.tpl'))
+
+PYBOOST_NATIVE_VIEW_GRAD_FUNCTION_TEMPLATE = Template.load_from_file(
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/backward/op_grad/template/native_view_grad_function.tpl'))
 
 PYBOOST_NATIVE_GRAD_FUNCTIONS_TEMPLATE = Template.load_from_file(
     os.path.join(K.WORK_DIR,
-                 './mindspore/ccsrc/pynative/grad/function/template/pyboost_native_grad_functions.tpl'))
+                 './mindspore/ccsrc/pynative/backward/op_grad/template/pyboost_native_grad_functions.tpl'))
 
 PYBOOST_NATIVE_GRAD_FUNCTIONS_HEADER_TEMPLATE = Template.load_from_file(
     os.path.join(K.WORK_DIR,
-                 './mindspore/ccsrc/pynative/grad/function/template/pyboost_native_grad_functions_header.tpl'))
+                 './mindspore/ccsrc/pynative/backward/op_grad/template/pyboost_native_grad_functions_header.tpl'))
 
 GEN_OPS_DEF_HEADER_TEMPLATE = Template.load_from_file(
     os.path.join(K.WORK_DIR, './mindspore/python/mindspore/ops_generate/op_def/gen_ops_def_header.tpl'))
@@ -281,33 +230,33 @@ PYBOOST_OP_REGISTER_TEMPLATE = Template.load_from_file(
 # Ascend op generate
 PYBOOST_ASCEND_OP_HEADER_TEMPLATE = Template.load_from_file(
     os.path.join(K.WORK_DIR,
-                 f'./{K.MS_OPS_KERNEL_PATH}/ascend/pyboost/template/pyboost_aclnn_header_template.tpl'))
+                 f'./{K.MS_OPS_KERNEL_PATH}/ascend/aclnn/pyboost_impl/template/pyboost_aclnn_header_template.tpl'))
 PYBOOST_ASCEND_INTERNAL_OP_HEADER_TEMPLATE = Template.load_from_file(
-    os.path.join(K.WORK_DIR,
-                 f'./{K.MS_OPS_KERNEL_PATH}/ascend/pyboost/internal/template/pyboost_internal_header_template.tpl')
+    os.path.join(K.WORK_DIR, K.MS_OPS_KERNEL_PATH,
+                 f'ascend/aclnn/pyboost_impl/internal/template/pyboost_internal_header_template.tpl')
 )
 
 PYBOOST_ASCEND_OP_SOURCE_TEMPLATE = Template.load_from_file(
     os.path.join(K.WORK_DIR,
-                 f'./{K.MS_OPS_KERNEL_PATH}/ascend/pyboost/template/pyboost_aclnn_source_template.tpl'))
+                 f'./{K.MS_OPS_KERNEL_PATH}/ascend/aclnn/pyboost_impl/template/pyboost_aclnn_source_template.tpl'))
 
 PYBOOST_ASCEND_SINGLE_OP_HEADER_TEMPLATE = Template.load_from_file(
-    os.path.join(K.WORK_DIR,
-                 f'./{K.MS_OPS_KERNEL_PATH}/ascend/pyboost/template/pyboost_aclnn_single_op_header_template.tpl'))
+    os.path.join(K.WORK_DIR, K.MS_OPS_KERNEL_PATH,
+                 f'ascend/aclnn/pyboost_impl/template/pyboost_aclnn_single_op_header_template.tpl'))
 
 PYBOOST_ASCEND_SINGLE_HCLL_OP_HEADER_TEMPLATE = Template.load_from_file(
-    os.path.join(K.WORK_DIR,
-                 f'./{K.MS_OPS_KERNEL_PATH}/ascend/pyboost/template/pyboost_aclnn_single_hccl_op_header_template.tpl'))
+    os.path.join(K.WORK_DIR, K.MS_OPS_KERNEL_PATH,
+                 f'ascend/aclnn/pyboost_impl/template/pyboost_aclnn_single_hccl_op_header_template.tpl'))
 
 PYBOOST_CALL_FUNC_TEMPLATE = Template('${return_type} Call(${call_args_with_type}) override;')
 
 PYBOOST_ASCEND_SINGLE_OP_SOURCE_TEMPLATE = Template.load_from_file(
-    os.path.join(K.WORK_DIR,
-                 f'./{K.MS_OPS_KERNEL_PATH}/ascend/pyboost/template/pyboost_aclnn_single_op_source_template.tpl'))
+    os.path.join(K.WORK_DIR, K.MS_OPS_KERNEL_PATH,
+                 f'ascend/aclnn/pyboost_impl/template/pyboost_aclnn_single_op_source_template.tpl'))
 
 PYBOOST_ASCEND_CALL_TEMPLATE = Template.load_from_file(
     os.path.join(K.WORK_DIR,
-                 f'./{K.MS_OPS_KERNEL_PATH}/ascend/pyboost/template/pyboost_ascend_call_template.tpl'))
+                 f'./{K.MS_OPS_KERNEL_PATH}/ascend/aclnn/pyboost_impl/template/pyboost_ascend_call_template.tpl'))
 
 PYBOOST_ASCEND_VIEW_CALL_TEMPLATE = Template.load_from_file(
     os.path.join(K.WORK_DIR,
@@ -316,7 +265,7 @@ PYBOOST_ASCEND_VIEW_CALL_TEMPLATE = Template.load_from_file(
 
 PYBOOST_ASCEND_CUSTOMIZE_CALL_TEMPLATE = Template.load_from_file(
     os.path.join(K.WORK_DIR,
-                 f'./{K.MS_OPS_KERNEL_PATH}/ascend/pyboost/template'
+                 f'./{K.MS_OPS_KERNEL_PATH}/ascend/aclnn/pyboost_impl/template'
                  '/pyboost_ascend_customize_call_template.tpl'))
 
 # GPU op generate
@@ -390,30 +339,6 @@ def ${func_name}(${func_args}):
     \"\"\"
     return ${func_impl_name}_impl(${input_args})\n\n""")
 
-OP_PROTO_TEMPLATE = Template("""
-${func_impl_declaration}
-OpDef g${class_name} = {
-  /*.name_=*/"${class_name}",
-  /*.args_=*/ {
-    ${input_args}
-  },
-  /* .returns_ = */ {
-    ${return_args}
-  },
-  /*.signatures_ =*/ {
-    ${signatures}
-  },
-  /*.indexes_ =*/ {
-    ${indexes}
-  },
-  /*.func_impl_=*/${func_impl_define},
-  /*.enable_dispatch_ =*/${enable_dispatch},
-  /*.is_view_ =*/${is_view},
-  /*.is_graph_view_ =*/${is_graph_view},
-};
-REGISTER_PRIMITIVE_OP_DEF(${class_name}, &g${class_name});
-""")
-
 MULTI_OUTPUT_TEMPLATE = """
 ValuePtrList values;
 (void)std::transform(op->outputs().begin(), op->outputs().end(), std::back_inserter(values),
@@ -421,86 +346,6 @@ ValuePtrList values;
 auto output_value = std::make_shared<ValueTuple>(values);
 """
 
-PY_LICENSE_STR = f"""# Copyright 2023 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.
-# ============================================================================
-"""
-
-OPS_PY_PRIM_HEADER = f"""
-\"\"\"Operators definition generated by gen_ops.py, includes primitive classes.\"\"\"
-
-from mindspore.ops.primitive import Primitive, prim_arg_register
-from mindspore.ops import signature as sig
-from mindspore.common import dtype as mstype
-from mindspore.common._decorator import deprecated
-from mindspore.ops._primitive_cache import _get_cache_prim
-from mindspore.ops._utils.arg_dtype_cast import type_it
-from mindspore.ops._utils.arg_handler import *
-from mindspore._c_expression import OpDtype
-from mindspore.common.jit_context import jit_context
-from mindspore._checkparam import is_stub_tensor
-"""
-
-OP_PRIM_CLASS_DEFINE_TEMPLATE = Template("""
-
-class ${class_name}(Primitive):
-${class_desc}${signature_code}${deprecated_code}
-${init_method}
-
-${call_method}""")
-
-OPS_PY_DEF_HEADER = f"""
-\"\"\"Operators definition generated by gen_ops.py, includes functions.\"\"\"
-
-from .gen_ops_prim import *
-from .pyboost_inner_prim import *
-from mindspore.ops.operations.manually_defined.ops_def import *
-from mindspore.ops._primitive_cache import _get_cache_prim
-"""
-
-PRIMITIVE_CLASS_DESC = """    r\"\"\"
-    .. code-block::
-
-        prim = ops.$class_name($init_args_str)
-        out = prim($input_args_str)
-
-    is equivalent to
-
-    .. code-block::
-
-        ops.$func_name($args_str)
-
-    Refer to :func:`mindspore.ops.$func_name` for more details.
-    \"\"\"
-"""
-
-CC_LICENSE_STR = f"""/**
- * Copyright 2023-2025 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.
- */"""
-
 arg_default_value = Template("""
 \"\"\"Operator labels and args default value.\"\"\"
 from mindspore.common import dtype as mstype
@@ -539,10 +384,10 @@ op_template = Template("""void ${op_name}::set_${arg_name}(const ${dtype} &${arg
                        """             { return GetValue<${dtype}>(GetAttr("${arg_name}")); }\n\n""")
 
 ACLNN_KERNEL_CC_TEMPLATE = Template.load_from_file(
-    os.path.join(K.WORK_DIR, f'./{K.MS_OPS_KERNEL_PATH}/ascend/opapi/template/aclnn_kernel_cc.tpl'))
+    os.path.join(K.WORK_DIR, f'./{K.MS_OPS_KERNEL_PATH}/ascend/aclnn/kernel_mod_impl/template/aclnn_kernel_cc.tpl'))
 
 ACLNN_KERNEL_H_TEMPLATE = Template.load_from_file(
-    os.path.join(K.WORK_DIR, f'./{K.MS_OPS_KERNEL_PATH}/ascend/opapi/template/aclnn_kernel_h.tpl'))
+    os.path.join(K.WORK_DIR, f'./{K.MS_OPS_KERNEL_PATH}/ascend/aclnn/kernel_mod_impl/template/aclnn_kernel_h.tpl'))
 
 update_output_shape_and_size_template = Template("""
 void ${kernelmod_name}::UpdateOutputShapeAndSize(const std::vector<KernelTensor *> &,
@@ -562,39 +407,51 @@ TUPLE_TENSOR_NOT_SUPPORTED = Template("""
 
 FUNCTIONAL_MAP_CC_TEMPLATE = Template.load_from_file(
     os.path.join(K.WORK_DIR,
-                 f'./mindspore/ccsrc/pynative/op_function/template/functional_map_cc.tpl'))
+                 f'./mindspore/ccsrc/pynative/forward/pyboost/template/functional_map_cc.tpl'))
 
 FUNCTIONAL_MAP_H_TEMPLATE = Template.load_from_file(
     os.path.join(K.WORK_DIR,
-                 f'./mindspore/ccsrc/pynative/op_function/template/functional_map_h.tpl'))
+                 f'./mindspore/ccsrc/pynative/forward/pyboost/template/functional_map_h.tpl'))
 
 ADD_TENSOR_DOCS_TEMPLATE = Template.load_from_file(
-    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/op_function/template/tensor_docs_py.tpl'))
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/tensor_docs_py.tpl'))
 
 AUTO_GRAD_IMPL_CC_TEMPLATE = Template.load_from_file(
-    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/op_function/template/auto_grad_impl_cc.tpl'))
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/auto_grad_impl_cc.tpl'))
 
 AUTO_GRAD_REG_H_TEMPLATE = Template.load_from_file(
-    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/op_function/template/auto_grad_reg_h.tpl'))
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/auto_grad_reg_h.tpl'))
 
 FUNCTIONS_CC_TEMPLATE = Template.load_from_file(
-    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/op_function/template/functions_cc.tpl'))
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/functions_cc.tpl'))
 
 FUNCTION_BODY_TEMPLATE = Template.load_from_file(
-    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/op_function/template/function_body.tpl'))
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/function_body.tpl'))
+
+FUNCTION_VIEW_BODY_TEMPLATE = Template.load_from_file(
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/function_view_body.tpl'))
+
+FUNCTION_VIEW_CUSTOMIZE_BODY_TEMPLATE = Template.load_from_file(
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/function_view_customize_body.tpl'))
 
 FUNCTION_COMM_BODY_TEMPLATE = Template.load_from_file(
-    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/op_function/template/comm_function_body.tpl'))
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/comm_function_body.tpl'))
 
 FUNCTIONS_H_TEMPLATE = Template.load_from_file(
-    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/op_function/template/functions_h.tpl'))
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/functions_h.tpl'))
 
 DO_GRAD_FUNCTION_BODY_TEMPLATE = Template.load_from_file(
-    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/op_function/template/do_grad_function.tpl'))
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/do_grad_function.tpl'))
+
+DO_VIEW_GRAD_FUNCTION_BODY_TEMPLATE = Template.load_from_file(
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/do_view_grad_function.tpl'))
+
+DO_VIEW_CUSTOMIZE_GRAD_FUNCTION_BODY_TEMPLATE = Template.load_from_file(
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pynative/forward/pyboost/template/do_view_customize_grad_function.tpl'))
 
 TENSOR_PY_CC_TEMPLATE = Template.load_from_file(
-    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pybind_api/ir/template/tensor_py_gen.tpl'))
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pybind_api/ir/tensor/template/tensor_py_gen.tpl'))
 TENSOR_PY_H_TEMPLATE = Template.load_from_file(
-    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pybind_api/ir/template/tensor_py_genH.tpl'))
+    os.path.join(K.WORK_DIR, './mindspore/ccsrc/pybind_api/ir/tensor/template/tensor_py_genH.tpl'))
 OP_DEF_INC_HEAD_TEMPLATE = Template(
     "#include \"mindspore/ops/op_def/auto_generate/gen_ops_primitive_${prefix_char}.h\"")