提交 e11183b2 编写于 作者: W WeibiaoYu

Fix issue: Profiler not read the actual device id in training

上级 e1d627ff
...@@ -64,10 +64,22 @@ class Profiler: ...@@ -64,10 +64,22 @@ class Profiler:
def __init__(self, subgraph='all', is_detail=True, is_show_op_path=False, output_path='./data', 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=""): 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: if not dev_id:
dev_id = "0" dev_id = "0"
logger.error("Fail to get DEVICE_ID, use 0 instead.") logger.error("Fail to get DEVICE_ID, use 0 instead.")
self._dev_id = dev_id self._dev_id = dev_id
self._container_path = os.path.join(self._base_profiling_container_path, dev_id) self._container_path = os.path.join(self._base_profiling_container_path, dev_id)
data_path = os.path.join(self._container_path, "data") data_path = os.path.join(self._container_path, "data")
...@@ -88,7 +100,7 @@ class Profiler: ...@@ -88,7 +100,7 @@ class Profiler:
except ImportError: except ImportError:
logger.error("Profiling: fail to import context from mindspore.") logger.error("Profiling: fail to import context from mindspore.")
except ValueError as err: 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['AICPU_PROFILING_MODE'] = 'true'
os.environ['PROFILING_DIR'] = str(self._container_path) os.environ['PROFILING_DIR'] = str(self._container_path)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册