From e11183b2d691a768c1987ec90d6111a6dddd7174 Mon Sep 17 00:00:00 2001 From: WeibiaoYu Date: Thu, 28 May 2020 04:22:40 -0400 Subject: [PATCH] Fix issue: Profiler not read the actual device id in training --- mindinsight/profiler/profiling.py | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/mindinsight/profiler/profiling.py b/mindinsight/profiler/profiling.py index b7385f0..c0ac6d7 100644 --- a/mindinsight/profiler/profiling.py +++ b/mindinsight/profiler/profiling.py @@ -64,10 +64,22 @@ class Profiler: def __init__(self, subgraph='all', is_detail=True, is_show_op_path=False, output_path='./data', optypes_to_deal='', optypes_not_deal='Variable', job_id=""): - dev_id = os.getenv('DEVICE_ID') + + # get device_id + try: + import mindspore.context as context + dev_id = context.get_context("device_id") + except ImportError: + logger.error("Profiling: fail to import context from mindspore.") + except ValueError as err: + logger.error("Profiling: fail to get context %s", err.message) + + if not dev_id: + dev_id = os.getenv('DEVICE_ID') if not dev_id: dev_id = "0" logger.error("Fail to get DEVICE_ID, use 0 instead.") + self._dev_id = dev_id self._container_path = os.path.join(self._base_profiling_container_path, dev_id) data_path = os.path.join(self._container_path, "data") @@ -88,7 +100,7 @@ class Profiler: except ImportError: logger.error("Profiling: fail to import context from mindspore.") except ValueError as err: - logger.err("Profiling: fail to set context", err.message) + logger.error("Profiling: fail to set context, %s", err.message) os.environ['AICPU_PROFILING_MODE'] = 'true' os.environ['PROFILING_DIR'] = str(self._container_path) -- GitLab