Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MindSpore
mindspore
提交
c2ff5e3f
M
mindspore
项目概览
MindSpore
/
mindspore
通知
35
Star
15
Fork
15
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
1
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
M
mindspore
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
1
Issue
1
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
c2ff5e3f
编写于
9月 03, 2020
作者:
M
mindspore-ci-bot
提交者:
Gitee
9月 03, 2020
浏览文件
操作
浏览文件
下载
差异文件
!5707 [MS][LITE][Develop]fix ScheduleNode and fill parser
Merge pull request !5707 from sunsuodong/fix_fill_parser
上级
735f9858
f978beb1
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
12 addition
and
1 deletion
+12
-1
mindspore/lite/src/scheduler.cc
mindspore/lite/src/scheduler.cc
+11
-1
mindspore/lite/src/scheduler.h
mindspore/lite/src/scheduler.h
+1
-0
未找到文件。
mindspore/lite/src/scheduler.cc
浏览文件 @
c2ff5e3f
...
...
@@ -231,7 +231,7 @@ kernel::LiteKernel *Scheduler::ScheduleNode(const std::vector<tensor::Tensor *>
const
std
::
vector
<
tensor
::
Tensor
*>
&
out_tensors
,
const
mindspore
::
lite
::
PrimitiveC
*
primitive
,
const
schema
::
CNode
*
cnode
)
{
MS_ASSERT
(
nullptr
!=
primitive
);
auto
data_type
=
in_tensors
.
front
()
->
data_type
(
);
TypeId
data_type
=
GetFirstFp32Fp16OrInt8Type
(
in_tensors
);
kernel
::
KernelKey
desc
{
kernel
::
KERNEL_ARCH
::
kCPU
,
data_type
,
static_cast
<
schema
::
PrimitiveType
>
(
primitive
->
Type
())};
if
(
context_
->
device_ctx_
.
type
==
DT_GPU
)
{
desc
.
arch
=
kernel
::
KERNEL_ARCH
::
kGPU
;
...
...
@@ -271,6 +271,16 @@ kernel::LiteKernel *Scheduler::ScheduleNode(const std::vector<tensor::Tensor *>
return
nullptr
;
}
TypeId
Scheduler
::
GetFirstFp32Fp16OrInt8Type
(
const
std
::
vector
<
tensor
::
Tensor
*>
&
in_tensors
)
{
for
(
const
auto
&
tensor
:
in_tensors
)
{
auto
dtype
=
tensor
->
data_type
();
if
(
dtype
==
kNumberTypeFloat32
||
dtype
==
kNumberTypeFloat16
||
dtype
==
kNumberTypeInt8
)
{
return
dtype
;
}
}
return
kNumberTypeFloat32
;
}
void
Scheduler
::
SetKernelTensorDataType
(
kernel
::
LiteKernel
*
kernel
)
{
if
(
kernel
->
desc
().
arch
!=
kernel
::
KERNEL_ARCH
::
kCPU
)
{
return
;
...
...
mindspore/lite/src/scheduler.h
浏览文件 @
c2ff5e3f
...
...
@@ -47,6 +47,7 @@ class Scheduler {
void
ConstructSubgraphs
(
std
::
vector
<
kernel
::
LiteKernel
*>
*
kernels
);
kernel
::
LiteKernel
*
CreateSubKernel
(
const
std
::
vector
<
kernel
::
LiteKernel
*>
&
kernels
,
kernel
::
KERNEL_ARCH
arch
);
TypeId
GetFirstFp32Fp16OrInt8Type
(
const
std
::
vector
<
tensor
::
Tensor
*>
&
in_tensors
);
void
SetKernelTensorDataType
(
kernel
::
LiteKernel
*
kernel
);
protected:
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录