Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
magicwindyyd
mindspore
提交
a948fe05
M
mindspore
项目概览
magicwindyyd
/
mindspore
与 Fork 源项目一致
Fork自
MindSpore / mindspore
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
M
mindspore
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
a948fe05
编写于
8月 19, 2020
作者:
X
xuyongfei
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
serving: update build.sh to add minddata support for acl
上级
3259dafa
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
5 addition
and
5 deletion
+5
-5
build.sh
build.sh
+0
-4
mindspore/ccsrc/backend/session/infer_session.cc
mindspore/ccsrc/backend/session/infer_session.cc
+5
-1
未找到文件。
build.sh
浏览文件 @
a948fe05
...
@@ -392,11 +392,7 @@ build_mindspore()
...
@@ -392,11 +392,7 @@ build_mindspore()
if
[[
-n
"
$VERBOSE
"
]]
;
then
if
[[
-n
"
$VERBOSE
"
]]
;
then
CMAKE_VERBOSE
=
"--verbose"
CMAKE_VERBOSE
=
"--verbose"
fi
fi
if
[[
"X
$ENABLE_ACL
"
=
"Xon"
]]
;
then
cmake
--build
.
${
CMAKE_VERBOSE
}
-j
$THREAD_NUM
else
cmake
--build
.
--target
package
${
CMAKE_VERBOSE
}
-j
$THREAD_NUM
cmake
--build
.
--target
package
${
CMAKE_VERBOSE
}
-j
$THREAD_NUM
fi
echo
"success to build mindspore project!"
echo
"success to build mindspore project!"
}
}
...
...
mindspore/ccsrc/backend/session/infer_session.cc
浏览文件 @
a948fe05
...
@@ -141,7 +141,7 @@ Status ServingTensor2MSTensor(size_t index, const InferTensorBase &out_tensor, t
...
@@ -141,7 +141,7 @@ Status ServingTensor2MSTensor(size_t index, const InferTensorBase &out_tensor, t
}
}
ms_tensor
=
std
::
make_shared
<
tensor
::
Tensor
>
(
data_type
,
shape
);
ms_tensor
=
std
::
make_shared
<
tensor
::
Tensor
>
(
data_type
,
shape
);
if
(
ms_tensor
->
Size
()
!=
out_tensor
.
data_size
())
{
if
(
out_tensor
.
data_size
()
==
0
||
ms_tensor
->
Size
()
!=
out_tensor
.
data_size
())
{
MSI_LOG_ERROR
<<
"input "
<<
std
::
to_string
(
index
)
MSI_LOG_ERROR
<<
"input "
<<
std
::
to_string
(
index
)
<<
" data size not match shape and dtype, calculated required size "
<<
ms_tensor
->
Size
()
<<
" data size not match shape and dtype, calculated required size "
<<
ms_tensor
->
Size
()
<<
", given "
<<
out_tensor
.
data_size
();
<<
", given "
<<
out_tensor
.
data_size
();
...
@@ -149,6 +149,10 @@ Status ServingTensor2MSTensor(size_t index, const InferTensorBase &out_tensor, t
...
@@ -149,6 +149,10 @@ Status ServingTensor2MSTensor(size_t index, const InferTensorBase &out_tensor, t
<<
" data size not match shape and dtype, calculated required size "
<<
" data size not match shape and dtype, calculated required size "
<<
ms_tensor
->
Size
()
<<
", given "
<<
out_tensor
.
data_size
();
<<
ms_tensor
->
Size
()
<<
", given "
<<
out_tensor
.
data_size
();
}
}
if
(
out_tensor
.
data
()
==
nullptr
||
ms_tensor
->
data_c
()
==
nullptr
)
{
MSI_LOG_ERROR
<<
"invalid data buffer"
;
return
FAILED
;
}
memcpy_s
(
ms_tensor
->
data_c
(),
ms_tensor
->
Size
(),
out_tensor
.
data
(),
out_tensor
.
data_size
());
memcpy_s
(
ms_tensor
->
data_c
(),
ms_tensor
->
Size
(),
out_tensor
.
data
(),
out_tensor
.
data_size
());
return
SUCCESS
;
return
SUCCESS
;
}
}
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录