Browse Source

!11931 bugfix: profiler mem usage: added warning log when running on GPU

From: @zhangyunshu
Reviewed-by: @yelihua
Signed-off-by:
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
a6e7b9c6e6
1 changed files with 6 additions and 0 deletions
  1. +6
    -0
      mindspore/profiler/profiling.py

+ 6
- 0
mindspore/profiler/profiling.py View File

@@ -269,6 +269,12 @@ class Profiler:

os.environ['PROFILING_MODE'] = str("false")

logger.warning(
'\nMemory Usage is not supported on GPU currently.\n'
'Please running on Ascend if you would like to see memory analysis, '
'otherwise, this warning can be ignored.'
)

def _analyse_step_trace(self, source_path=None, framework_parser=None, is_training_mode_flag=True):
"""
Analyse step trace data and save the result.


Loading…
Cancel
Save