Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MindSpore
mindinsight
提交
c19da6a9
M
mindinsight
项目概览
MindSpore
/
mindinsight
通知
7
Star
3
Fork
2
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
M
mindinsight
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
c19da6a9
编写于
7月 27, 2020
作者:
L
Li Hongzhang
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix dsmi_get_device_utilization_rate
上级
8fb96dd9
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
9 addition
and
9 deletion
+9
-9
mindinsight/sysmetric/collector/_collect_npu.py
mindinsight/sysmetric/collector/_collect_npu.py
+9
-9
未找到文件。
mindinsight/sysmetric/collector/_collect_npu.py
浏览文件 @
c19da6a9
...
...
@@ -270,8 +270,7 @@ def dsmi_get_hbm_info(device_id):
}
@
_timeout
(
0.2
,
0
)
@
_fallback_to_prev_result
@
_timeout
(
0.2
,
-
1
)
def
dsmi_get_device_utilization_rate
(
device_id
,
device_type
):
"""
Get device utilization rate, %.
...
...
@@ -282,13 +281,14 @@ def dsmi_get_device_utilization_rate(device_id, device_type):
device_id (int): The specific device id
device_type (int): The device type, 1 for memory, 2 AI Core, 5 memory bandwidth, 6 HBM, 10 HBM bandwidth.
Returns:
int, the utilization rate.
int, the utilization rate
, returning -1 to indicate querying failed
.
"""
device_id
=
c_int
(
device_id
)
device_type
=
c_int
(
device_type
)
utilization_rate
=
c_uint
()
success
=
_libsmicall
(
device_id
,
device_type
,
byref
(
utilization_rate
))
return
success
,
utilization_rate
.
value
if
_libsmicall
(
device_id
,
device_type
,
byref
(
utilization_rate
)):
return
utilization_rate
.
value
return
-
1
@
_fallback_to_prev_result
...
...
@@ -388,14 +388,14 @@ def _collect_one(device_id):
Raises:
RuntimeError, when querying dsmi returning non-zero.
"""
kb_to_mb
,
memory_threshold
,
success
=
1024
,
4
,
[
True
]
*
7
kb_to_mb
,
memory_threshold
,
success
=
1024
,
4
,
[
True
]
*
6
success
[
0
],
health
=
dsmi_get_device_health
(
device_id
)
success
[
1
],
hbm_info
=
dsmi_get_hbm_info
(
device_id
)
success
[
2
],
chip_info
=
dsmi_get_chip_info
(
device_id
)
success
[
3
],
ip_addr
=
dsmi_get_device_ip_address
(
device_id
)
success
[
4
],
aicore_rate
=
dsmi_get_device_utilization_rate
(
device_id
,
2
)
success
[
5
],
power_info
=
dsmi_get_device_power_info
(
device_id
)
success
[
6
],
temperature
=
dsmi_get_device_temperature
(
device_id
)
success
[
4
],
power_info
=
dsmi_get_device_power_info
(
device_id
)
success
[
5
],
temperature
=
dsmi_get_device_temperature
(
device_id
)
aicore_rate
=
dsmi_get_device_utilization_rate
(
device_id
,
2
)
return
{
'chip_name'
:
chip_info
.
get
(
'chip_name'
),
'device_id'
:
device_id
,
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录