diff --git a/mindspore/profiler/profiling.py b/mindspore/profiler/profiling.py
index 8b4f970cde..361c1b9210 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")