!23982 fix the debug info about the cell function decorated by @ms_functon

Merge pull request !23982 from huanghui/fix-filename-error-with-ms_function

mindspore/_extends/parse/parser.py

@@ -535,7 +535,7 @@ class Parser:
         self.fn = fn
         self.parse_method = parse_method
         self.line_offset = 0
-        self.filename: str = inspect.getfile(self.fn)
+        self.filename: str = inspect.getfile(inspect.unwrap(self.fn))
 
         # Used to resolve mindspore builtin ops namespace.
         self.ms_common_ns = CellNamespace('mindspore.common')

tests/ut/python/pipeline/parse/test_debug_info.py

@@ -0,0 +1,71 @@
+# Copyright 2021 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.
+# ============================================================================
+""" test_parse_numpy """
+import os
+import shutil
+import subprocess
+import numpy as np
+import mindspore as ms
+from mindspore import nn
+from mindspore import context
+from mindspore import ms_function
+from mindspore import Tensor
+from tests.security_utils import security_off_wrap
+
+
+def find_files(file, para):
+    output = subprocess.check_output(
+        ["grep '%s' %s | wc -l" % (para, file)],
+        shell=True)
+    out = str(output, 'utf-8').strip()
+    return out
+
+
+def remove_path(path):
+    if os.path.exists(path):
+        shutil.rmtree(path)
+
+
+@security_off_wrap
+def test_ms_function():
+    @ms_function
+    def add(x):
+        return x + 1
+
+    context.set_context(mode=context.GRAPH_MODE)
+    context.set_context(save_graphs=True, save_graphs_path="ir_dump_path")
+    input1 = np.random.randn(5, 5)
+    add(Tensor(input1, ms.float32))
+    result = find_files("./ir_dump_path/*validate*.ir", "test_debug_info.py(45)/        return x + 1/")
+    assert result == '2'
+    remove_path("./ir_dump_path/")
+
+
+@security_off_wrap
+def test_cell_ms_function():
+    class Net(nn.Cell):
+
+        @ms_function
+        def construct(self, x):
+            return x
+
+    context.set_context(mode=context.GRAPH_MODE)
+    context.set_context(save_graphs=True, save_graphs_path="ir_dump_path")
+    input1 = np.random.randn(5, 5)
+    net = Net()
+    net(Tensor(input1, ms.float32))
+    result = find_files("./ir_dump_path/*validate*.ir", "test_debug_info.py(62)/            return x/")
+    assert result == '1'
+    remove_path("./ir_dump_path/")