mindspore 2.3.0__cp39-cp39-win_amd64.whl → 2.4.1__cp39-cp39-win_amd64.whl

mindspore/profiler/__init__.py

@@ -20,8 +20,9 @@ Users can visualize the results using the MindInsight tool.
 Now, Profiler supports AICORE operator, AICPU operator, HostCPU operator, memory,
 correspondence, cluster, etc data analysis.
 """
-__all__ = ["Profiler", "EnvProfiler", "ProfilerLevel"]
+__all__ = ["Profiler", "EnvProfiler", "ProfilerLevel", "DynamicProfilerMonitor"]
 
 from mindspore.profiler.profiling import Profiler
 from mindspore.profiler.envprofiling import EnvProfiler
 from mindspore.profiler.profiling import ProfilerLevel
+from mindspore.profiler.dynamic_profiler import DynamicProfilerMonitor

mindspore/profiler/common/constant.py

@@ -0,0 +1,29 @@
+# Copyright 2020-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.
+# ============================================================================
+"""Constant values"""
+from enum import Enum
+
+
+class DeviceTarget(Enum):
+    """The device target enum."""
+    CPU = 'cpu'
+    GPU = 'gpu'
+    ASCEND = 'ascend'
+
+
+class ProfilerLevel(Enum):
+    Level0 = "Level0"
+    Level1 = "Level1"
+    Level2 = "Level2"

mindspore/profiler/common/registry.py

@@ -0,0 +1,47 @@
+# Copyright 2020-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.
+# ============================================================================
+"""CPU platform profiler."""
+from typing import Any, Dict
+from mindspore import log as logger
+
+
+class Register:
+    """
+    Common register class
+    """
+    def __init__(self, name) -> None:
+        self.module_dict = dict()
+        self.name = name
+
+    def register_module(self, name):
+        """ registers module decorator."""
+        def decorator(target):
+            model_name = target.__name__
+            if name in self.module_dict:
+                raise Exception(f'{model_name} already registered with name: {name}')
+
+            self.module_dict[name] = target
+            logger.info(f'registered module: {model_name} with name: {name}')
+            return target
+
+        return decorator
+
+    @property
+    def modules(self) -> Dict[str, Any]:
+        """ get all registered modules """
+        return self.module_dict
+
+
+PROFILERS = Register('PROFILERS')

mindspore/profiler/common/util.py

@@ -18,6 +18,8 @@ Profiler util.
 This module provides the utils.
 """
 import os
+import time
+from functools import wraps
 
 # one sys count takes 10 ns, 1 ms has 100000 system count
 import re
@@ -27,6 +29,32 @@ import stat
 from mindspore import log as logger
 
 
+def timeit(custom_message=None):
+    """
+    Decorator to measure the execution time of a function.
+
+    Args:
+        custom_message (str, optional): Custom message to display. If not provided, the function name will be used.
+
+    Returns:
+        The result of the decorated function.
+    """
+    def decorator(func):
+        @wraps(func)
+        def wrapper(*args, **kwargs):
+            start_time = time.perf_counter()
+            result = func(*args, **kwargs)
+            execution_time = time.perf_counter() - start_time
+
+            message = custom_message or func.__name__
+            print(f"[{os.getpid()}] {message}, cost time {execution_time:.2f} s", flush=True)
+
+            return result
+
+        return wrapper
+
+    return decorator
+
 def to_int(param, param_name):
     """
     Transfer param to int type.