From 4dff1d6831316f1d2576b9958c6d8e5007405ab1 Mon Sep 17 00:00:00 2001 From: zhangyihui Date: Tue, 15 Sep 2020 21:23:37 +0800 Subject: [PATCH] fix the problem: it is not reasonable to need special setting aicpu can be used --- mindspore/profiler/profiling.py | 1 + 1 file changed, 1 insertion(+) diff --git a/mindspore/profiler/profiling.py b/mindspore/profiler/profiling.py index 273a61a4d5..6819477c21 100644 --- a/mindspore/profiler/profiling.py +++ b/mindspore/profiler/profiling.py @@ -107,6 +107,7 @@ class Profiler: os.environ['PROFILING_MODE'] = 'true' os.environ['MINDDATA_PROFILING_DIR'] = self._output_path os.environ['DEVICE_ID'] = self._dev_id + os.environ['AICPU_PROFILING_MODE'] = 'true' # use context interface to open profiling, for the new mindspore version(after 2020.5.21) context.set_context(enable_profiling=True, profiling_options="training_trace:task_trace")