forked from mindspore-Ecosystem/mindspore
profiler mem usage: added a warning prompt in log when running on GPU
This commit is contained in:
parent
882eee75e3
commit
092945e310
|
@ -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.
|
||||
|
|
Loading…
Reference in New Issue