profiler mem usage: added a warning prompt in log when running on GPU

This commit is contained in:
zhangyunshu 2021-02-01 11:20:02 +08:00
parent 882eee75e3
commit 092945e310
1 changed files with 6 additions and 0 deletions

View File

@ -281,6 +281,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.