|
|
@ -107,6 +107,7 @@ class Profiler:
|
|
|
|
os.environ['PROFILING_MODE'] = 'true'
|
|
|
|
os.environ['PROFILING_MODE'] = 'true'
|
|
|
|
os.environ['MINDDATA_PROFILING_DIR'] = self._output_path
|
|
|
|
os.environ['MINDDATA_PROFILING_DIR'] = self._output_path
|
|
|
|
os.environ['DEVICE_ID'] = self._dev_id
|
|
|
|
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)
|
|
|
|
# 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")
|
|
|
|
context.set_context(enable_profiling=True, profiling_options="training_trace:task_trace")
|
|
|
|