mindstudio-probe 8.3.2__py3-none-any.whl → 26.0.0a1__py3-none-any.whl

msprobe/msaccucmp/format_manager/builtin_format_convert/convert_HWCN_to_FRACTAL_Z.py

@@ -0,0 +1,89 @@
+# coding=utf-8
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
+#
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
+#
+#          http://license.coscl.org.cn/MulanPSL2
+#
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
+
+"""
+Function:
+convert format from HWCN to FRACTAL_Z.
+"""
+from itertools import product
+import numpy as np
+
+from msprobe.msaccucmp.cmp_utils.constant.const_manager import ConstManager
+from msprobe.msaccucmp.cmp_utils.utils import least_common_multiple as lcm
+from msprobe.msaccucmp.cmp_utils.utils import ceiling_divide as ceil
+
+
+def _get_axis(ghw_axis: list, value_map: dict, dst_c: int, w_axis: int) -> int:
+    g_axis = ghw_axis[0]
+    h_axis = ghw_axis[1]
+    kh_axis = value_map.get('kh_axis')
+    kw_axis = value_map.get('kw_axis')
+    e_multi = value_map.get('e_multi')
+    tmp_value = (g_axis // e_multi) * (dst_c // ConstManager.C0_AXIS) * kh_axis * kw_axis
+    return tmp_value + (dst_c // ConstManager.C0_AXIS) * kh_axis * kw_axis + h_axis * kw_axis + w_axis
+
+
+def _get_count_for_axis(shape_from: list, g_num: int, e_multi: int) -> int:
+    kh_axis = shape_from[0]
+    kw_axis = shape_from[1]
+    c_ori = shape_from[2]
+    c_opt = ceil(e_multi * c_ori, ConstManager.C0_AXIS) * ConstManager.C0_AXIS
+    c1_axis = ceil(c_opt, ConstManager.C0_AXIS)
+    return g_num * c1_axis * kh_axis * kw_axis
+
+
+def convert(shape_from: list, shape_to: list, array: any, group: int = 1) -> any:
+    """
+    Convert the data format from HWCN to FRACTAL_Z
+    :param shape_from: the shape before convert
+    :param shape_to: the shape after convert
+    :param array: the one-dimensional array
+    :param group: group for group_conv, default value is 1
+    :return: the data array of FRACTAL_Z shape
+    """
+    _ = shape_to
+    # get before convert shape
+    kh_axis = shape_from[0]
+    kw_axis = shape_from[1]
+    c_ori = shape_from[2]
+    array_shape = array.reshape(kh_axis, kw_axis, c_ori, shape_from[3])
+
+    # CUBE Unit K and N
+    n_ori = shape_from[3] // group
+
+    # Specific multiplying algorithm to get e_multi
+    e_multi = min(lcm(lcm(c_ori, ConstManager.C0_AXIS) // c_ori, lcm(n_ori, ConstManager.N0_AXIS) // n_ori), group)
+    array_to = np.zeros(
+        (_get_count_for_axis(shape_from, ceil(group, e_multi), e_multi),
+         ceil(ceil(e_multi * n_ori, ConstManager.N0_AXIS) * ConstManager.N0_AXIS, ConstManager.N0_AXIS),
+         ConstManager.N0_AXIS,
+         ConstManager.C0_AXIS), dtype=array_shape.dtype)
+    # convert hwcn to gc1hwn1n0c0
+    for g_axis, h_axis, w_axis, c_axis, n_axis in \
+            product(range(group), range(kh_axis), range(kw_axis), range(c_ori), range(n_ori)):
+        e_val = g_axis % e_multi
+        dst_c = e_val * c_ori + c_axis
+        dst_n = e_val * n_ori + n_axis
+        src_n = g_axis * n_ori + n_axis
+        array_to[_get_axis([g_axis, h_axis, w_axis],
+                            {'e_multi': e_multi, 'kh_axis': kh_axis, 'kw_axis': kw_axis},
+                            dst_c, w_axis)][dst_n // ConstManager.N0_AXIS][
+            dst_n % ConstManager.N0_AXIS][dst_c % ConstManager.C0_AXIS] = \
+            array_shape[h_axis][w_axis][c_axis][src_n]
+    return array_to
+
+

msprobe/msaccucmp/format_manager/builtin_format_convert/convert_HWCN_to_NCHW.py

@@ -0,0 +1,37 @@
+# coding=utf-8
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
+#
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
+#
+#          http://license.coscl.org.cn/MulanPSL2
+#
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
+
+"""
+Function:
+convert format from HWCN to NCHW.
+"""
+
+
+def convert(shape_from: list, shape_to: list, array: any) -> any:
+    """
+    Convert the data format from HWCN to NCHW
+    :param shape_from: the shape before convert
+    :param shape_to: the shape after convert
+    :param array: the one-dimensional array
+    :return:the data array of NCHW shape
+    """
+    _ = shape_to
+    h_from = shape_from[0]
+    w_from = shape_from[1]
+    c_from = shape_from[2]
+    n_from = shape_from[3]
+    return array.reshape(h_from, w_from, c_from, n_from).transpose(3, 2, 0, 1)

msprobe/msaccucmp/format_manager/builtin_format_convert/convert_HWCN_to_NHWC.py

@@ -0,0 +1,37 @@
+# coding=utf-8
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
+#
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
+#
+#          http://license.coscl.org.cn/MulanPSL2
+#
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
+
+"""
+Function:
+convert format from HWCN to NHWC.
+"""
+
+
+def convert(shape_from: list, shape_to: list, array: any) -> any:
+    """
+    Convert the data format from HWCN to NHWC
+    :param shape_from: the shape before convert
+    :param shape_to: the shape after convert
+    :param array: the one-dimensional array
+    :return: the data array of NHWC shape
+    """
+    _ = shape_to
+    h_from = shape_from[0]
+    w_from = shape_from[1]
+    c_from = shape_from[2]
+    n_from = shape_from[3]
+    return array.reshape(h_from, w_from, c_from, n_from).transpose(3, 0, 1, 2)

msprobe/msaccucmp/format_manager/builtin_format_convert/convert_NC1HWC0_to_HWCN.py

@@ -0,0 +1,43 @@
+# coding=utf-8
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
+#
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
+#
+#          http://license.coscl.org.cn/MulanPSL2
+#
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
+
+"""
+Function:
+convert format from NC1HWC0 to HWCN.
+"""
+import numpy as np
+
+
+def convert(shape_from: list, shape_to: list, array: any) -> any:
+    """
+    Convert the data format from NC1HWC0 to HWCN
+    :param shape_from: the shape before convert
+    :param shape_to: the shape after convert
+    :param array: the one-dimensional array
+    :return: the data array of HWCN shape
+    """
+    _ = shape_to
+    n_from = shape_from[0]
+    c1_from = shape_from[1]
+    h_from = shape_from[2]
+    w_from = shape_from[3]
+    c0_from = shape_from[4]
+
+    array_shape = array.reshape(n_from, c1_from, h_from, w_from, c0_from)
+    tmp_input_tensor = np.transpose(array_shape, axes=(2, 3, 1, 4, 0))
+    tmp_input_tensor = tmp_input_tensor.reshape((h_from, w_from, c1_from * c0_from, n_from))
+    return tmp_input_tensor

msprobe/msaccucmp/format_manager/builtin_format_convert/convert_NC1HWC0_to_NCHW.py

@@ -0,0 +1,48 @@
+# coding=utf-8
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
+#
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
+#
+#          http://license.coscl.org.cn/MulanPSL2
+#
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
+
+"""
+Function:
+convert format from NC1HWC0 to NCHW.
+"""
+import numpy as np
+from msprobe.msaccucmp.cmp_utils import log
+from msprobe.msaccucmp.cmp_utils.constant.compare_error import CompareError
+
+
+def convert(shape_from: list, shape_to: list, array: any) -> any:
+    """
+    Convert the data format from NC1HWC0 to NCHW
+    :param shape_from: the shape before convert
+    :param shape_to: the shape after convert
+    :param array: the one-dimensional array
+    :return: the data array of NCHW shape
+    """
+    _ = shape_to
+    if len(shape_from) < 5:
+        log.print_error_log("length of shape of NC1HWC0 is less than 5, please check.")
+        raise CompareError(CompareError.MSACCUCMP_INDEX_OUT_OF_BOUNDS_ERROR)
+    n_from = shape_from[0]
+    c1_from = shape_from[1]
+    h_from = shape_from[2]
+    w_from = shape_from[3]
+    c0_from = shape_from[4]
+
+    array_shape = array.reshape(n_from, c1_from, h_from, w_from, c0_from)
+    tmp_input_tensor = np.transpose(array_shape, axes=(0, 1, 4, 2, 3))
+    tmp_input_tensor = tmp_input_tensor.reshape((n_from, c1_from * c0_from, h_from, w_from))
+    return tmp_input_tensor

msprobe/msaccucmp/format_manager/builtin_format_convert/convert_NC1HWC0_to_NHWC.py

@@ -0,0 +1,43 @@
+# coding=utf-8
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
+#
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
+#
+#          http://license.coscl.org.cn/MulanPSL2
+#
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
+
+"""
+Function:
+convert format from NC1HWC0 to NHWC.
+"""
+import numpy as np
+
+
+def convert(shape_from: list, shape_to: list, array: any) -> any:
+    """
+    Convert the data format from NC1HWC0 to NHWC
+    :param shape_from: the shape before convert
+    :param shape_to: the shape after convert
+    :param array: the one-dimensional array
+    :return: the data array of NHWC shape
+    """
+    _ = shape_to
+    n_from = shape_from[0]
+    c1_from = shape_from[1]
+    h_from = shape_from[2]
+    w_from = shape_from[3]
+    c0_from = shape_from[4]
+
+    array_shape = array.reshape(n_from, c1_from, h_from, w_from, c0_from)
+    tmp_input_tensor = np.transpose(array_shape, axes=(0, 2, 3, 1, 4))
+    tmp_input_tensor = tmp_input_tensor.reshape((n_from, h_from, w_from, c1_from * c0_from))
+    return tmp_input_tensor

msprobe/msaccucmp/format_manager/builtin_format_convert/convert_NCHW_to_FRACTAL_Z.py

@@ -0,0 +1,87 @@
+# coding=utf-8
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
+#
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
+#
+#          http://license.coscl.org.cn/MulanPSL2
+#
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
+
+"""
+Function:
+convert format from NCHW to FRACTAL_Z.
+"""
+from itertools import product
+import numpy as np
+
+from msprobe.msaccucmp.cmp_utils.constant.const_manager import ConstManager
+from msprobe.msaccucmp.cmp_utils.utils import least_common_multiple as lcm
+from msprobe.msaccucmp.cmp_utils.utils import ceiling_divide as ceil
+
+
+def _get_axis(gnc_axis: list, value_map: dict, dst_c: int, w_axis: int, h_axis: int) -> int:
+    g_axis = gnc_axis[0]
+    kh_axis = value_map.get('kh_axis')
+    kw_axis = value_map.get('kw_axis')
+    e_multi = value_map.get('e_multi')
+    tmp_value = (g_axis // e_multi) * (dst_c // ConstManager.C0_AXIS) * kh_axis * kw_axis
+    return tmp_value + (dst_c // ConstManager.C0_AXIS) * kh_axis * kw_axis + h_axis * kw_axis + w_axis
+
+
+def _get_count_for_axis(shape_from: list, g_num: int, e_multi: int) -> int:
+    c_ori = shape_from[1]
+    kh_axis = shape_from[2]
+    kw_axis = shape_from[3]
+    c_opt = ceil(e_multi * c_ori, ConstManager.C0_AXIS) * ConstManager.C0_AXIS
+    c1_axis = ceil(c_opt, ConstManager.C0_AXIS)
+    return g_num * c1_axis * kh_axis * kw_axis
+
+
+def convert(shape_from: list, shape_to: list, array: any, group: int = 1) -> any:
+    """
+    Convert the data format from NCHW to FRACTAL_Z
+    :param shape_from: the shape before convert
+    :param shape_to: the shape after convert
+    :param array: the one-dimensional array
+    :param group: group for group_conv, default value is 1
+    :return: the data array of FRACTAL_Z shape
+    """
+    _ = shape_to
+    # get before convert shape
+    c_ori = shape_from[1]
+    kh_axis = shape_from[2]
+    kw_axis = shape_from[3]
+    array_shape = array.reshape(shape_from[0], c_ori, kh_axis, kw_axis)
+
+    # CUBE Unit K and N
+    n_ori = shape_from[0] // group
+
+    # Specific multiplying algorithm to get e_multi
+    e_multi = min(lcm(lcm(c_ori, ConstManager.C0_AXIS) // c_ori, lcm(n_ori, ConstManager.N0_AXIS) // n_ori), group)
+    array_to = np.zeros(
+        (_get_count_for_axis(shape_from, ceil(group, e_multi), e_multi),
+         ceil(ceil(e_multi * n_ori, ConstManager.N0_AXIS) * ConstManager.N0_AXIS, ConstManager.N0_AXIS),
+         ConstManager.N0_AXIS,
+         ConstManager.C0_AXIS), dtype=array_shape.dtype)
+    # convert nchw to gc1hwn1n0c0
+    for g_axis, n_axis, c_axis, h_axis, w_axis in \
+            product(range(group), range(n_ori), range(c_ori), range(kh_axis), range(kw_axis)):
+        e_val = g_axis % e_multi
+        dst_c = e_val * c_ori + c_axis
+        dst_n = e_val * n_ori + n_axis
+        src_n = g_axis * n_ori + n_axis
+        array_to[_get_axis([g_axis, n_axis, c_axis],
+                            {'e_multi': e_multi, 'kh_axis': kh_axis, 'kw_axis': kw_axis},
+                            dst_c, w_axis, h_axis)][dst_n // ConstManager.N0_AXIS][
+            dst_n % ConstManager.N0_AXIS][dst_c % ConstManager.C0_AXIS] = \
+            array_shape[src_n][c_axis][h_axis][w_axis]
+    return array_to
+

msprobe/msaccucmp/format_manager/builtin_format_convert/convert_NCHW_to_NHWC.py

@@ -0,0 +1,37 @@
+# coding=utf-8
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
+#
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
+#
+#          http://license.coscl.org.cn/MulanPSL2
+#
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
+
+"""
+Function:
+convert format from NCHW to NHWC.
+"""
+
+
+def convert(shape_from: list, shape_to: list, array: any) -> any:
+    """
+    Convert the data format from NCHW to NHWC
+    :param shape_from: the shape before convert
+    :param shape_to: the shape after convert
+    :param array: the one-dimensional array
+    :return: the data array of NHWC shape
+    """
+    _ = shape_to
+    n_from = shape_from[0]
+    c_from = shape_from[1]
+    h_from = shape_from[2]
+    w_from = shape_from[3]
+    return array.reshape(n_from, c_from, h_from, w_from).transpose(0, 2, 3, 1)

msprobe/msaccucmp/format_manager/builtin_format_convert/convert_NDC1HWC0_to_NCDHW.py

@@ -0,0 +1,48 @@
+# coding=utf-8
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
+#
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
+#
+#          http://license.coscl.org.cn/MulanPSL2
+#
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
+
+"""
+Function:
+convert format from NDC1HWC0 to NCDHW.
+"""
+import numpy as np
+from msprobe.msaccucmp.cmp_utils import log
+from msprobe.msaccucmp.cmp_utils.constant.compare_error import CompareError
+
+
+def convert(shape_from: list, shape_to: list, array: any) -> any:
+    """
+    Convert the data format from NDC1HWC0 to NCDHW
+    :param shape_from: the shape before convert
+    :param shape_to: the shape after convert
+    :param array: the one-dimensional array
+    :return:the data array of NCDHW shape
+    """
+    if len(shape_from) < 6:
+        log.print_error_log("length of shape of NDC1HWC0 is less than 6, please check.")
+        raise CompareError(CompareError.MSACCUCMP_INDEX_OUT_OF_BOUNDS_ERROR)
+    axis_n = shape_from[0]
+    axis_d = shape_from[1]
+    axis_c1 = shape_from[2]
+    axis_h = shape_from[3]
+    axis_w = shape_from[4]
+    axis_c0 = shape_from[5]
+    c_pad = None if axis_c1 * axis_c0 == shape_to[1] else shape_to[1] - axis_c1 * axis_c0
+    tmp_array = array.reshape(shape_from)
+    tmp_array = np.transpose(tmp_array, axes=(0, 2, 5, 1, 3, 4))
+    tmp_array = tmp_array.reshape((axis_n, axis_c1 * axis_c0, axis_d, axis_h, axis_w))
+    return tmp_array[:, :c_pad, :, :, :]

msprobe/msaccucmp/format_manager/builtin_format_convert/convert_NDC1HWC0_to_ND.py

@@ -0,0 +1,44 @@
+# coding=utf-8
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
+#
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
+#
+#          http://license.coscl.org.cn/MulanPSL2
+#
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
+
+"""
+Function:
+convert format from NDC1HWC0 to ND.
+"""
+import numpy as np
+
+
+def convert(shape_from: list, shape_to: list, array: any) -> any:
+    """
+    Convert the data format from NDC1HWC0 to ND. Current only support 5 dimensions ND
+    :param shape_from: the shape before convert
+    :param shape_to: the shape after convert
+    :param array: the one-dimensional array
+    :return:the data array of ND shape
+    """
+    axis_n, axis_d, axis_c1, axis_h, axis_w, axis_c0 = shape_from
+
+    tmp_array = array.reshape(shape_from)
+    tmp_array = np.transpose(tmp_array, axes=(0, 2, 5, 1, 3, 4))
+    # first convert to NCDHW
+    tmp_array = tmp_array.reshape((axis_n, axis_c1 * axis_c0, axis_d, axis_h, axis_w))
+
+    if len(shape_to) == 5:
+        c_pad = None if axis_c1 * axis_c0 == shape_to[1] else shape_to[1] - axis_c1 * axis_c0
+        return tmp_array[:, :c_pad, ...]
+    else:
+        return tmp_array

msprobe/msaccucmp/format_manager/builtin_format_convert/convert_NHWC_to_FRACTAL_Z.py

@@ -0,0 +1,87 @@
+# coding=utf-8
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
+#
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
+#
+#          http://license.coscl.org.cn/MulanPSL2
+#
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
+
+"""
+Function:
+convert format from NHWC to FRACTAL_Z.
+"""
+from itertools import product
+import numpy as np
+
+from msprobe.msaccucmp.cmp_utils.constant.const_manager import ConstManager
+from msprobe.msaccucmp.cmp_utils.utils import least_common_multiple as lcm
+from msprobe.msaccucmp.cmp_utils.utils import ceiling_divide as ceil
+
+
+def _get_axis(gnh_axis: list, value_map: dict, dst_c: int, w_axis: int) -> int:
+    g_axis = gnh_axis[0]
+    h_axis = gnh_axis[2]
+    kh_axis = value_map.get('kh_axis')
+    kw_axis = value_map.get('kw_axis')
+    e_multi = value_map.get('e_multi')
+    tmp_value = (g_axis // e_multi) * (dst_c // ConstManager.C0_AXIS) * kh_axis * kw_axis
+    return tmp_value + (dst_c // ConstManager.C0_AXIS) * kh_axis * kw_axis + h_axis * kw_axis + w_axis
+
+
+def _get_count_for_axis(shape_from: list, g_num: int, e_multi: int) -> int:
+    c_ori = shape_from[3]
+    kh_axis = shape_from[1]
+    kw_axis = shape_from[2]
+    c_opt = ceil(e_multi * c_ori, ConstManager.C0_AXIS) * ConstManager.C0_AXIS
+    c1_axis = ceil(c_opt, ConstManager.C0_AXIS)
+    return g_num * c1_axis * kh_axis * kw_axis
+
+
+def convert(shape_from: list, shape_to: list, array: any, group: int = 1) -> any:
+    """
+    Convert the data format from NHWC to FRACTAL_Z
+    :param shape_from: the shape before convert
+    :param shape_to: the shape after convert
+    :param array: the one-dimensional array
+    :param group: group for group_conv, default value is 1
+    :return: the data array of FRACTAL_Z shape
+    """
+    _ = shape_to
+    # get before convert shape
+    kh_axis = shape_from[1]
+    kw_axis = shape_from[2]
+    c_ori = shape_from[3]
+    array_shape = array.reshape(shape_from[0], kh_axis, kw_axis, c_ori)
+
+    # CUBE Unit K and N
+    n_ori = shape_from[0] // group
+
+    # Specific multiplying algorithm to get e_multi
+    e_multi = min(lcm(lcm(c_ori, ConstManager.C0_AXIS) // c_ori, lcm(n_ori, ConstManager.N0_AXIS) // n_ori), group)
+    array_to = np.zeros(
+        (_get_count_for_axis(shape_from, ceil(group, e_multi), e_multi),
+         ceil(ceil(e_multi * n_ori, ConstManager.N0_AXIS) * ConstManager.N0_AXIS, ConstManager.N0_AXIS),
+         ConstManager.N0_AXIS, ConstManager.C0_AXIS), dtype=array_shape.dtype)
+    # convert nhwc to gc1hwn1n0c0
+    for g_axis, n_axis, h_axis, w_axis, c_axis in \
+            product(range(group), range(n_ori), range(kh_axis), range(kw_axis), range(c_ori)):
+        e_val = g_axis % e_multi
+        dst_c = e_val * c_ori + c_axis
+        dst_n = e_val * n_ori + n_axis
+        src_n = g_axis * n_ori + n_axis
+        array_to[_get_axis([g_axis, n_axis, h_axis],
+                            {'e_multi': e_multi, 'kh_axis': kh_axis, 'kw_axis': kw_axis},
+                            dst_c, w_axis)][dst_n // ConstManager.N0_AXIS][
+            dst_n % ConstManager.N0_AXIS][dst_c % ConstManager.C0_AXIS] = \
+            array_shape[src_n][h_axis][w_axis][c_axis]
+    return array_to
+

msprobe/msaccucmp/format_manager/builtin_format_convert/convert_NHWC_to_HWCN.py

@@ -0,0 +1,37 @@
+# coding=utf-8
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
+#
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
+#
+#          http://license.coscl.org.cn/MulanPSL2
+#
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
+
+"""
+Function:
+convert format from NHWC to HWCN.
+"""
+
+
+def convert(shape_from: list, shape_to: list, array: any) -> any:
+    """
+    Convert the data format from NHWC to HWCN
+    :param shape_from: the shape before convert
+    :param shape_to: the shape after convert
+    :param array: the one-dimensional array
+    :return:the data array of HWCN shape
+    """
+    _ = shape_to
+    n_from = shape_from[0]
+    h_from = shape_from[1]
+    w_from = shape_from[2]
+    c_from = shape_from[3]
+    return array.reshape(n_from, h_from, w_from, c_from).transpose(1, 2, 3, 0)