Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
正统之独孤求败
mindspore
提交
236d6c6d
M
mindspore
项目概览
正统之独孤求败
/
mindspore
与 Fork 源项目一致
Fork自
MindSpore / mindspore
通知
1
Star
0
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看板
体验新版 GitCode,发现更多精彩内容 >>
提交
236d6c6d
编写于
6月 16, 2020
作者:
K
kswang
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix cpu reshape bug
上级
f10e2974
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
8 addition
and
3 deletion
+8
-3
mindspore/ccsrc/device/cpu/cpu_kernel_runtime.cc
mindspore/ccsrc/device/cpu/cpu_kernel_runtime.cc
+8
-3
未找到文件。
mindspore/ccsrc/device/cpu/cpu_kernel_runtime.cc
浏览文件 @
236d6c6d
...
...
@@ -161,8 +161,12 @@ BaseRef CPUKernelRuntime::CreatTensorForOutput(const AnfNodePtr &input_node, siz
}
tensor
::
TensorPtr
tensor
=
std
::
make_shared
<
tensor
::
Tensor
>
(
type_id
,
temp_shape
);
MS_EXCEPTION_IF_NULL
(
tensor
);
address
->
ptr_
=
tensor
->
data_c
(
true
);
address
->
ref_count_
=
INIT_NODE_REF
;
if
(
address
->
ref_count_
>
0
&&
address
->
ptr_
!=
nullptr
)
{
tensor
->
set_device_address
(
address
);
}
else
{
address
->
ptr_
=
tensor
->
data_c
(
true
);
address
->
ref_count_
=
INIT_NODE_REF
;
}
tensor
->
set_dirty
(
false
);
return
tensor
;
}
else
if
(
input_node
->
isa
<
Parameter
>
()
||
input_node
->
isa
<
ValueNode
>
())
{
...
...
@@ -211,6 +215,7 @@ void CPUKernelRuntime::BindInputOutput(const session::KernelGraph *kernel_graph,
}
tensor
->
set_dirty
(
true
);
}
address
->
ref_count_
=
INIT_NODE_REF
;
tensor
->
set_device_address
(
address
);
}
...
...
@@ -220,7 +225,7 @@ void CPUKernelRuntime::BindInputOutput(const session::KernelGraph *kernel_graph,
// new output and bind ptr
auto
output_nodes
=
kernel_graph
->
outputs
();
for
(
const
auto
&
item
:
output_nodes
)
{
auto
item_with_index
=
AnfAlgo
::
VisitKernelWithReturnType
(
item
,
0
);
auto
item_with_index
=
AnfAlgo
::
VisitKernelWithReturnType
(
item
,
0
,
true
);
auto
out
=
CreatTensorForOutput
(
item_with_index
.
first
,
item_with_index
.
second
,
input_map
);
outputs
->
push_back
(
std
::
move
(
out
));
}
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录