Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
OpenHarmony
Xts Acts
提交
3675f0b0
X
Xts Acts
项目概览
OpenHarmony
/
Xts Acts
1 年多 前同步成功
通知
9
Star
22
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
X
Xts Acts
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
3675f0b0
编写于
5月 19, 2023
作者:
O
openharmony_ci
提交者:
Gitee
5月 19, 2023
浏览文件
操作
浏览文件
下载
差异文件
!8680 适配新的NNRT API
Merge pull request !8680 from zhuguodong/master
上级
65dc38f3
d48dfe0b
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
15 addition
and
15 deletion
+15
-15
ai/mindspore/src/ohos_c_api_test_mslite.cpp
ai/mindspore/src/ohos_c_api_test_mslite.cpp
+15
-15
未找到文件。
ai/mindspore/src/ohos_c_api_test_mslite.cpp
浏览文件 @
3675f0b0
...
...
@@ -63,21 +63,21 @@ void AddContextDeviceCPU(OH_AI_ContextHandle context) {
// add nnrt device info
void
AddContextDeviceNNRT
(
OH_AI_ContextHandle
context
)
{
size_t
num
=
0
;
auto
desc
=
OH_AI_
NNRTGetAll
DeviceDescs
(
&
num
);
auto
desc
=
OH_AI_
GetAllNNRT
DeviceDescs
(
&
num
);
if
(
desc
==
nullptr
)
{
return
;
}
std
::
cout
<<
"found "
<<
num
<<
" nnrt devices"
<<
std
::
endl
;
auto
id
=
OH_AI_
NNRTDeviceDescGetId
(
desc
);
auto
name
=
OH_AI_
NNRTDeviceDescGetName
(
desc
);
auto
type
=
OH_AI_
NNRTDeviceDescGetType
(
desc
);
auto
id
=
OH_AI_
GetDeviceIdFromNNRTDeviceDesc
(
desc
);
auto
name
=
OH_AI_
GetNameFromNNRTDeviceDesc
(
desc
);
auto
type
=
OH_AI_
GetTypeFromNNRTDeviceDesc
(
desc
);
std
::
cout
<<
"NNRT device: id = "
<<
id
<<
", name: "
<<
name
<<
", type:"
<<
type
<<
std
::
endl
;
OH_AI_DeviceInfoHandle
nnrt_device_info
=
OH_AI_DeviceInfoCreate
(
OH_AI_DEVICETYPE_NNRT
);
ASSERT_NE
(
nnrt_device_info
,
nullptr
);
OH_AI_DeviceInfoSetDeviceId
(
nnrt_device_info
,
id
);
OH_AI_
NNRTDestroyAll
DeviceDescs
(
&
desc
);
OH_AI_
DestroyAllNNRT
DeviceDescs
(
&
desc
);
OH_AI_DeviceType
device_type
=
OH_AI_DeviceInfoGetDeviceType
(
nnrt_device_info
);
printf
(
"==========device_type:%d
\n
"
,
device_type
);
...
...
@@ -94,18 +94,18 @@ void AddContextDeviceNNRT(OH_AI_ContextHandle context) {
// add nnrt device info by type
void
AddContextDeviceNNRTByType
(
OH_AI_ContextHandle
context
)
{
size_t
num
=
0
;
auto
desc
=
OH_AI_
NNRTGetAll
DeviceDescs
(
&
num
);
auto
desc
=
OH_AI_
GetAllNNRT
DeviceDescs
(
&
num
);
if
(
desc
==
nullptr
)
{
return
;
}
std
::
cout
<<
"found "
<<
num
<<
" nnrt devices"
<<
std
::
endl
;
auto
id
=
OH_AI_
NNRTDeviceDescGetId
(
desc
);
auto
name
=
OH_AI_
NNRTDeviceDescGetName
(
desc
);
auto
type
=
OH_AI_
NNRTDeviceDescGetType
(
desc
);
auto
id
=
OH_AI_
GetDeviceIdFromNNRTDeviceDesc
(
desc
);
auto
name
=
OH_AI_
GetNameFromNNRTDeviceDesc
(
desc
);
auto
type
=
OH_AI_
GetTypeFromNNRTDeviceDesc
(
desc
);
std
::
cout
<<
"NNRT device: id = "
<<
id
<<
", name: "
<<
name
<<
", type:"
<<
type
<<
std
::
endl
;
auto
nnrt_device_info
=
OH_AI_CreateNNRTDeviceInfoByType
(
type
);
OH_AI_
NNRTDestroyAll
DeviceDescs
(
&
desc
);
OH_AI_
DestroyAllNNRT
DeviceDescs
(
&
desc
);
ASSERT_NE
(
nnrt_device_info
,
nullptr
);
OH_AI_DeviceType
device_type
=
OH_AI_DeviceInfoGetDeviceType
(
nnrt_device_info
);
...
...
@@ -119,18 +119,18 @@ void AddContextDeviceNNRTByType(OH_AI_ContextHandle context) {
// add nnrt device info by name
void
AddContextDeviceNNRTByName
(
OH_AI_ContextHandle
context
)
{
size_t
num
=
0
;
auto
desc
=
OH_AI_
NNRTGetAll
DeviceDescs
(
&
num
);
auto
desc
=
OH_AI_
GetAllNNRT
DeviceDescs
(
&
num
);
if
(
desc
==
nullptr
)
{
return
;
}
std
::
cout
<<
"found "
<<
num
<<
" nnrt devices"
<<
std
::
endl
;
auto
id
=
OH_AI_
NNRTDeviceDescGetId
(
desc
);
auto
name
=
OH_AI_
NNRTDeviceDescGetName
(
desc
);
auto
type
=
OH_AI_
NNRTDeviceDescGetType
(
desc
);
auto
id
=
OH_AI_
GetDeviceIdFromNNRTDeviceDesc
(
desc
);
auto
name
=
OH_AI_
GetNameFromNNRTDeviceDesc
(
desc
);
auto
type
=
OH_AI_
GetTypeFromNNRTDeviceDesc
(
desc
);
std
::
cout
<<
"NNRT device: id = "
<<
id
<<
", name: "
<<
name
<<
", type:"
<<
type
<<
std
::
endl
;
auto
nnrt_device_info
=
OH_AI_CreateNNRTDeviceInfoByName
(
name
);
OH_AI_
NNRTDestroyAll
DeviceDescs
(
&
desc
);
OH_AI_
DestroyAllNNRT
DeviceDescs
(
&
desc
);
ASSERT_NE
(
nnrt_device_info
,
nullptr
);
OH_AI_DeviceType
device_type
=
OH_AI_DeviceInfoGetDeviceType
(
nnrt_device_info
);
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录