mindstudio-probe 1.0.4__py3-none-any.whl → 1.1.0__py3-none-any.whl

msprobe/mindspore/dump/jit_dump.py

@@ -1,12 +1,30 @@
+# Copyright (c) 2024-2024, Huawei Technologies Co., Ltd.
+# All rights reserved.
+#
+# 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.
+
 import os
+from collections import defaultdict
 
 from mindspore import Tensor
-from mindspore.common.api import _MindsporeFunctionExecutor
 from mindspore._c_expression import PyNativeExecutor_
+from mindspore.common.api import _MindsporeFunctionExecutor
 
 from msprobe.mindspore.dump.hook_cell.api_registry import api_register
-from msprobe.core.data_dump.data_processor.base import ModuleForwardInputsOutputs
+from msprobe.core.data_dump.data_processor.base import ModuleForwardInputsOutputs, ModuleBackwardInputsOutputs
 from msprobe.core.common.const import Const
+from msprobe.core.data_dump.data_processor.base import ModuleForwardInputsOutputs
+from msprobe.mindspore.dump.hook_cell.api_registry import api_register
 
 
 def dump_jit(name, in_feat, out_feat, is_forward):
@@ -17,19 +35,27 @@ def dump_jit(name, in_feat, out_feat, is_forward):
         result = ori_args[0:index]
     else:
         result = "JitFunction"
-    if is_forward:
-        name_template = "Jit." + result + ".forward"
-    else:
-        name_template = "Jit." + result + ".backward"
     if JitDump.need_dump():
-        JitDump.data_collector.update_api_or_module_name(name_template)
-        module_input_output = ModuleForwardInputsOutputs(args=in_feat, kwargs={}, output=out_feat)
-        JitDump.data_collector.forward_data_collect(name_template, {}, pid, module_input_output)
+        if is_forward:
+            JitDump.jit_count[result] += 1
+            name_template = Const.JIT + Const.SEP + result + Const.SEP + str(JitDump.jit_count[result]) + Const.SEP + \
+                            Const.FORWARD
+            JitDump.data_collector.update_api_or_module_name(name_template)
+            module_input_output = ModuleForwardInputsOutputs(args=in_feat, kwargs={}, output=out_feat)
+            JitDump.data_collector.forward_data_collect(name_template, None, pid, module_input_output)
+        else:
+            name_template = Const.JIT + Const.SEP + result + Const.SEP + str(JitDump.jit_count[result]) + Const.SEP + \
+                            Const.BACKWARD
+            JitDump.data_collector.update_api_or_module_name(name_template)
+            module_input_output = ModuleBackwardInputsOutputs(grad_input=in_feat ,grad_output=out_feat)
+            JitDump.data_collector.backward_data_collect(name_template, None, pid, module_input_output)
 
 
 class JitDump(_MindsporeFunctionExecutor):
     dump_config = None
     jit_enable = False
+    jit_dump_switch = True
+    jit_count = defaultdict(int)
 
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
@@ -38,11 +64,9 @@ class JitDump(_MindsporeFunctionExecutor):
     def __call__(self, *args, **kwargs):
         api_register.api_set_ori_func()
         out = super().__call__(*args, **kwargs)
-        if isinstance(args[0], Tensor):
-            dump_jit({}, args, out, True)
-        else:
-            dump_jit(args[0], args[1:], out, True)
-        JitDump.jit_enable = True
+        if JitDump.jit_dump_switch and len(args) > 0:
+            dump_jit(args[0], args, out, True)
+            JitDump.jit_enable = True
         api_register.api_set_hook_func()
         return out
 
@@ -62,11 +86,11 @@ class JitDump(_MindsporeFunctionExecutor):
             return False
         return True
 
-    def grad(self, obj, grad, weights, grad_position, *args,  **kwargs):
-        if JitDump.jit_enable:
+    def grad(self, obj, grad, weights, grad_position, *args, **kwargs):
+        if JitDump.jit_dump_switch and JitDump.jit_enable:
             api_register.api_set_ori_func()
         output = self._executor.grad(grad, obj, weights, grad_position, *args, *(kwargs.values()))
-        if JitDump.jit_enable:
+        if JitDump.jit_dump_switch and JitDump.jit_enable:
             dump_jit(obj, args, None, False)
             api_register.api_set_hook_func()
         return output

msprobe/mindspore/dump/kernel_graph_dump.py

@@ -1,8 +1,24 @@
-import os
+# Copyright (c) 2024-2024, Huawei Technologies Co., Ltd.
+# All rights reserved.
+#
+# 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.
+
 import json
-from msprobe.mindspore.debugger.debugger_config import DebuggerConfig
-from msprobe.mindspore.common.log import logger
+import os
+
 from msprobe.core.common.file_utils import FileOpen, create_directory
+from msprobe.mindspore.common.log import logger
+from msprobe.mindspore.debugger.debugger_config import DebuggerConfig
 
 
 class KernelGraphDump:

msprobe/mindspore/dump/kernel_kbyk_dump.py

@@ -1,10 +1,25 @@
-import os
+# Copyright (c) 2024-2024, Huawei Technologies Co., Ltd.
+# All rights reserved.
+#
+# 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.
+
 import json
+import os
 
-from msprobe.mindspore.debugger.debugger_config import DebuggerConfig
-from msprobe.mindspore.common.log import logger
-from msprobe.core.common.file_utils import FileOpen, create_directory
 from msprobe.core.common.const import Const
+from msprobe.core.common.file_utils import FileOpen, create_directory
+from msprobe.mindspore.common.log import logger
+from msprobe.mindspore.debugger.debugger_config import DebuggerConfig
 
 
 class KernelKbykDump:

msprobe/mindspore/free_benchmark/api_pynative_self_check.py

@@ -1,17 +1,32 @@
-import os
-import inspect
+# Copyright (c) 2024-2024, Huawei Technologies Co., Ltd.
+# All rights reserved.
+#
+# 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.
+
 import importlib
+import inspect
+import os
 
 import mindspore as ms
 from mindspore.communication import comm_func
 
-from msprobe.core.common.file_utils import load_yaml, check_path_length
 from msprobe.core.common.const import Const
+from msprobe.core.common.file_utils import check_path_length, load_yaml
 from msprobe.mindspore.common.const import Const as MsConst
 from msprobe.mindspore.common.const import FreeBenchmarkConst
-from msprobe.mindspore.free_benchmark.common.config import Config
 from msprobe.mindspore.common.log import logger
 from msprobe.mindspore.debugger.debugger_config import DebuggerConfig
+from msprobe.mindspore.free_benchmark.common.config import Config
 from msprobe.mindspore.free_benchmark.decorator.decorator_factory import decorate_forward_function
 
 

msprobe/mindspore/free_benchmark/common/config.py

@@ -1,3 +1,18 @@
+# Copyright (c) 2024-2024, Huawei Technologies Co., Ltd.
+# All rights reserved.
+#
+# 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.
+
 from msprobe.mindspore.common.const import FreeBenchmarkConst
 
 

msprobe/mindspore/free_benchmark/common/handler_params.py

@@ -1,3 +1,18 @@
+# Copyright (c) 2024-2024, Huawei Technologies Co., Ltd.
+# All rights reserved.
+#
+# 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.
+
 from typing import Optional, Any, Tuple, Dict, Callable
 
 

msprobe/mindspore/free_benchmark/common/utils.py

@@ -1,14 +1,28 @@
-from typing import Any
-from typing import Optional
+# Copyright (c) 2024-2024, Huawei Technologies Co., Ltd.
+# All rights reserved.
+#
+# 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.
+
 from dataclasses import dataclass
+from typing import Any, Optional
 
 import mindspore as ms
 from mindspore import Tensor
 
-from msprobe.mindspore.runtime import Runtime
 from msprobe.mindspore.common.const import FreeBenchmarkConst
-from .config import Config
-from .handler_params import HandlerParams
+from msprobe.mindspore.free_benchmark.common.config import Config
+from msprobe.mindspore.free_benchmark.common.handler_params import HandlerParams
+from msprobe.mindspore.runtime import Runtime
 
 
 class Tools:

msprobe/mindspore/free_benchmark/decorator/dec_forward.py

@@ -1,6 +1,20 @@
+# Copyright (c) 2024-2024, Huawei Technologies Co., Ltd.
+# All rights reserved.
+#
+# 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.
+
+from msprobe.mindspore.common.const import Const, FreeBenchmarkConst
 from msprobe.mindspore.free_benchmark.common.config import Config
-from msprobe.mindspore.common.const import Const
-from msprobe.mindspore.common.const import FreeBenchmarkConst
 from msprobe.mindspore.free_benchmark.common.handler_params import HandlerParams
 from msprobe.mindspore.free_benchmark.handler.handler_factory import HandlerFactory
 from msprobe.mindspore.free_benchmark.perturbation.perturbation_factory import PerturbationFactory

msprobe/mindspore/free_benchmark/decorator/decorator_factory.py

@@ -1,16 +1,31 @@
+# Copyright (c) 2024-2024, Huawei Technologies Co., Ltd.
+# All rights reserved.
+#
+# 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.
+
 import os
 import sys
 import traceback
 from functools import wraps
-from typing import Tuple, Dict, List
+from typing import Dict, List, Tuple
 
 from mindspore import ops
 
-from msprobe.mindspore.runtime import Runtime
 from msprobe.mindspore.common.log import logger
 from msprobe.mindspore.free_benchmark.common.config import Config
 from msprobe.mindspore.free_benchmark.common.handler_params import HandlerParams
-from .dec_forward import ForwardSelfChecker
+from msprobe.mindspore.free_benchmark.decorator.dec_forward import ForwardSelfChecker
+from msprobe.mindspore.runtime import Runtime
 
 
 def decorate(original_func, decorate_func, api_name=None):

msprobe/mindspore/free_benchmark/handler/base_handler.py

@@ -1,14 +1,29 @@
+# Copyright (c) 2024-2024, Huawei Technologies Co., Ltd.
+# All rights reserved.
+#
+# 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.
+
 import math
 from abc import ABC, abstractmethod
-from typing import Any, Tuple, Optional
+from typing import Any, Optional, Tuple
 
 import mindspore as ms
 from mindspore import Tensor, ops
 
-from msprobe.mindspore.common.log import logger
-from msprobe.mindspore.free_benchmark.common.utils import Tools
 from msprobe.mindspore.common.const import FreeBenchmarkConst
+from msprobe.mindspore.common.log import logger
 from msprobe.mindspore.free_benchmark.common.handler_params import HandlerParams
+from msprobe.mindspore.free_benchmark.common.utils import Tools
 
 
 class BaseHandler(ABC):

msprobe/mindspore/free_benchmark/handler/check_handler.py

@@ -1,14 +1,29 @@
-from typing import Any
+# Copyright (c) 2024-2024, Huawei Technologies Co., Ltd.
+# All rights reserved.
+#
+# 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.
+
 from dataclasses import asdict
+from typing import Any
 
 from mindspore import Tensor, ops
 
+from msprobe.core.data_dump.json_writer import DataWriter
 from msprobe.mindspore.common.log import logger
 from msprobe.mindspore.free_benchmark.common.config import Config
-from msprobe.mindspore.free_benchmark.handler.base_handler import BaseHandler
 from msprobe.mindspore.free_benchmark.common.handler_params import HandlerParams
 from msprobe.mindspore.free_benchmark.common.utils import make_unequal_row
-from msprobe.core.data_dump.json_writer import DataWriter
+from msprobe.mindspore.free_benchmark.handler.base_handler import BaseHandler
 
 
 class CheckHandler(BaseHandler):

msprobe/mindspore/free_benchmark/handler/fix_handler.py

@@ -1,3 +1,18 @@
+# Copyright (c) 2024-2024, Huawei Technologies Co., Ltd.
+# All rights reserved.
+#
+# 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.
+
 from typing import Any
 
 from mindspore import Tensor

msprobe/mindspore/free_benchmark/handler/handler_factory.py

@@ -1,8 +1,23 @@
+# Copyright (c) 2024-2024, Huawei Technologies Co., Ltd.
+# All rights reserved.
+#
+# 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.
+
+from msprobe.mindspore.common.const import FreeBenchmarkConst
 from msprobe.mindspore.common.log import logger
 from msprobe.mindspore.free_benchmark.common.config import Config
-from msprobe.mindspore.common.const import FreeBenchmarkConst
-from .check_handler import CheckHandler
-from .fix_handler import FixHandler
+from msprobe.mindspore.free_benchmark.handler.check_handler import CheckHandler
+from msprobe.mindspore.free_benchmark.handler.fix_handler import FixHandler
 
 
 class HandlerFactory:

msprobe/mindspore/free_benchmark/perturbation/add_noise.py

@@ -1,11 +1,26 @@
+# Copyright (c) 2024-2024, Huawei Technologies Co., Ltd.
+# All rights reserved.
+#
+# 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.
+
 from typing import Any
 
 from mindspore import Tensor, ops
 
+from msprobe.mindspore.common.const import FreeBenchmarkConst
 from msprobe.mindspore.common.log import logger
-from msprobe.mindspore.free_benchmark.perturbation.base_perturbation import BasePerturbation
 from msprobe.mindspore.free_benchmark.common.handler_params import HandlerParams
-from msprobe.mindspore.common.const import FreeBenchmarkConst
+from msprobe.mindspore.free_benchmark.perturbation.base_perturbation import BasePerturbation
 
 
 class AddNoisePerturbation(BasePerturbation):
@@ -43,25 +58,25 @@ class AddNoisePerturbation(BasePerturbation):
 
         return inputs
 
-    def _get_noise(self, input):
+    def _get_noise(self, tensor):
         """
         得到要添加的噪声值
 
         """
         if self.is_fuzzed:
             return False
-        if not ops.is_floating_point(input) or ops.numel(input) == 0:
+        if not ops.is_floating_point(tensor) or ops.numel(tensor) == 0:
             return False
 
-        pert_value = FreeBenchmarkConst.PERT_VALUE_DICT.get(input.dtype)
+        pert_value = FreeBenchmarkConst.PERT_VALUE_DICT.get(tensor.dtype)
         if not pert_value:
             return False
         else:
             self.perturbation_value = pert_value
 
-        max_val = ops.max(ops.abs(input))[0].item()
+        max_val = ops.max(ops.abs(tensor))[0].item()
         if max_val < pert_value:
             return False
 
-        noise = ops.full(input.shape, self.perturbation_value, dtype=input.dtype)
+        noise = ops.full(tensor.shape, self.perturbation_value, dtype=tensor.dtype)
         return noise

msprobe/mindspore/free_benchmark/perturbation/base_perturbation.py

@@ -1,3 +1,18 @@
+# Copyright (c) 2024-2024, Huawei Technologies Co., Ltd.
+# All rights reserved.
+#
+# 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.
+
 from typing import Any
 
 from msprobe.mindspore.free_benchmark.common.handler_params import HandlerParams

msprobe/mindspore/free_benchmark/perturbation/bit_noise.py

@@ -1,10 +1,25 @@
+# Copyright (c) 2024-2024, Huawei Technologies Co., Ltd.
+# All rights reserved.
+#
+# 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.
+
 from typing import Any
 
 import numpy as np
 from mindspore import Tensor, ops
 
-from msprobe.mindspore.common.log import logger
 from msprobe.mindspore.common.const import FreeBenchmarkConst
+from msprobe.mindspore.common.log import logger
 from msprobe.mindspore.free_benchmark.common.handler_params import HandlerParams
 from msprobe.mindspore.free_benchmark.perturbation.base_perturbation import BasePerturbation
 
@@ -45,19 +60,19 @@ class BitNoisePerturbation(BasePerturbation):
         params.args = args
         return self.get_fuzzed_result(params)
 
-    def _get_bit_len_type(self, input):
+    def _get_bit_len_type(self, tensor):
         if self.is_fuzzed:
             return False
-        if not isinstance(input, Tensor) or not ops.is_floating_point(input) or \
-           input.numel() == 0:
+        if not isinstance(tensor, Tensor) or not ops.is_floating_point(tensor) or \
+           tensor.numel() == 0:
             return False
-        bit_len_type = FreeBenchmarkConst.PERT_BIT_DICT.get(input.dtype)
+        bit_len_type = FreeBenchmarkConst.PERT_BIT_DICT.get(tensor.dtype)
         if not bit_len_type:
             return False
-        pert_value = FreeBenchmarkConst.PERT_VALUE_DICT.get(input.dtype)
+        pert_value = FreeBenchmarkConst.PERT_VALUE_DICT.get(tensor.dtype)
         if not pert_value:
             return False
-        max_val = ops.max(ops.abs(input))[0].item()
+        max_val = ops.max(ops.abs(tensor))[0].item()
         if max_val < pert_value:
             return False
         return bit_len_type

msprobe/mindspore/free_benchmark/perturbation/exchange_value.py

@@ -1,14 +1,39 @@
+# Copyright (c) 2024-2024, Huawei Technologies Co., Ltd.
+# All rights reserved.
+#
+# 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.
+
 from typing import Any
 
-from mindspore import Tensor
+from mindspore import Tensor, ops
 
 from msprobe.mindspore.common.log import logger
-from msprobe.mindspore.free_benchmark.perturbation.base_perturbation import BasePerturbation
 from msprobe.mindspore.free_benchmark.common.handler_params import HandlerParams
+from msprobe.mindspore.free_benchmark.perturbation.base_perturbation import BasePerturbation
 
 
 class ExchangeValuePerturbation(BasePerturbation):
 
+    @staticmethod
+    def _check_tensor_shape(inputs):
+        dims = len(inputs.shape)
+        if dims == 1 and inputs.shape[0] > 1:
+            return True
+        if dims > 1 and inputs.shape[1] > 0:
+            if inputs.shape[0] > 1 or inputs.shape[1] > 1:
+                return True
+        return False
+
     def handle(self, params: HandlerParams) -> Any:
         """
         返回首尾交换后的api输出
@@ -25,22 +50,23 @@ class ExchangeValuePerturbation(BasePerturbation):
         返回首尾交换后的api输入
 
         """
-        if isinstance(inputs, Tensor):
-            if not self.is_fuzzed and len(inputs.shape) > 0 and inputs.shape[0] > 1:
-                result = inputs.copy()
-                if len(inputs.shape) == 1:
-                    first_element = inputs[0]
-                    last_element = inputs[-1]
-                    result[0] = last_element
-                    result[-1] = first_element
-                else:
-                    first_element = inputs[0][0]
-                    last_element = inputs[-1][-1]
-                    result[0][0] = last_element
-                    result[-1][-1] = first_element
-
-                self.is_fuzzed = True
-                return result
+        if isinstance(inputs, Tensor) and ops.is_floating_point(inputs):
+            if self.is_fuzzed or not self._check_tensor_shape(inputs):
+                return inputs
+            result = inputs.copy()
+            if len(inputs.shape) == 1:
+                first_element = inputs[0]
+                last_element = inputs[-1]
+                result[0] = last_element
+                result[-1] = first_element
+            else:
+                first_element = inputs[0][0]
+                last_element = inputs[-1][-1]
+                result[0][0] = last_element
+                result[-1][-1] = first_element
+
+            self.is_fuzzed = True
+            return result
 
         if isinstance(inputs, dict):
             return {k: self.exchange_value(v) for k, v in inputs.items()}