Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
正统之独孤求败
mindspore
提交
3fc89750
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,发现更多精彩内容 >>
提交
3fc89750
编写于
8月 20, 2020
作者:
M
mindspore-ci-bot
提交者:
Gitee
8月 20, 2020
浏览文件
操作
浏览文件
下载
差异文件
!4829 fix op_parameter free
Merge pull request !4829 from hangq/master
上级
8ea8c267
fd6556e6
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
12 addition
and
7 deletion
+12
-7
mindspore/lite/src/kernel_registry.cc
mindspore/lite/src/kernel_registry.cc
+8
-0
mindspore/lite/src/kernel_registry.h
mindspore/lite/src/kernel_registry.h
+2
-1
mindspore/lite/src/lite_kernel.h
mindspore/lite/src/lite_kernel.h
+2
-6
未找到文件。
mindspore/lite/src/kernel_registry.cc
浏览文件 @
3fc89750
...
@@ -108,6 +108,7 @@ kernel::LiteKernel *KernelRegistry::GetKernel(const std::vector<tensor::Tensor *
...
@@ -108,6 +108,7 @@ kernel::LiteKernel *KernelRegistry::GetKernel(const std::vector<tensor::Tensor *
<<
schema
::
EnumNamePrimitiveType
((
schema
::
PrimitiveType
)
primitive
->
Type
());
<<
schema
::
EnumNamePrimitiveType
((
schema
::
PrimitiveType
)
primitive
->
Type
());
return
nullptr
;
return
nullptr
;
}
}
this
->
op_parameters_
.
emplace_back
(
parameter
);
auto
creator
=
GetCreator
(
key
);
auto
creator
=
GetCreator
(
key
);
if
(
creator
!=
nullptr
)
{
if
(
creator
!=
nullptr
)
{
auto
kernel
=
creator
(
in_tensors
,
out_tensors
,
parameter
,
ctx
,
key
,
primitive
);
auto
kernel
=
creator
(
in_tensors
,
out_tensors
,
parameter
,
ctx
,
key
,
primitive
);
...
@@ -115,4 +116,11 @@ kernel::LiteKernel *KernelRegistry::GetKernel(const std::vector<tensor::Tensor *
...
@@ -115,4 +116,11 @@ kernel::LiteKernel *KernelRegistry::GetKernel(const std::vector<tensor::Tensor *
}
}
return
nullptr
;
return
nullptr
;
}
}
KernelRegistry
::~
KernelRegistry
()
{
for
(
auto
op_parameter
:
op_parameters_
)
{
delete
(
op_parameter
);
}
op_parameters_
.
clear
();
}
}
// namespace mindspore::lite
}
// namespace mindspore::lite
mindspore/lite/src/kernel_registry.h
浏览文件 @
3fc89750
...
@@ -32,7 +32,7 @@ namespace mindspore::lite {
...
@@ -32,7 +32,7 @@ namespace mindspore::lite {
class
KernelRegistry
{
class
KernelRegistry
{
public:
public:
KernelRegistry
()
=
default
;
KernelRegistry
()
=
default
;
virtual
~
KernelRegistry
()
=
default
;
virtual
~
KernelRegistry
();
static
KernelRegistry
*
GetInstance
();
static
KernelRegistry
*
GetInstance
();
int
Init
();
int
Init
();
...
@@ -54,6 +54,7 @@ class KernelRegistry {
...
@@ -54,6 +54,7 @@ class KernelRegistry {
static
const
int
op_type_length_
{
PrimitiveType_MAX
-
PrimitiveType_MIN
+
1
};
static
const
int
op_type_length_
{
PrimitiveType_MAX
-
PrimitiveType_MIN
+
1
};
static
const
int
array_size_
{
device_type_length_
*
data_type_length_
*
op_type_length_
};
static
const
int
array_size_
{
device_type_length_
*
data_type_length_
*
op_type_length_
};
kernel
::
KernelCreator
creator_arrays_
[
array_size_
]
=
{
0
};
kernel
::
KernelCreator
creator_arrays_
[
array_size_
]
=
{
0
};
std
::
vector
<
OpParameter
*>
op_parameters_
;
};
};
class
KernelRegistrar
{
class
KernelRegistrar
{
...
...
mindspore/lite/src/lite_kernel.h
浏览文件 @
3fc89750
...
@@ -57,6 +57,7 @@ struct KernelKey {
...
@@ -57,6 +57,7 @@ struct KernelKey {
class
LiteKernel
{
class
LiteKernel
{
public:
public:
LiteKernel
()
=
default
;
LiteKernel
()
=
default
;
// parameter should be deleted or freed by caller, and should be deleted or freed after LiteKernel is deleted
LiteKernel
(
OpParameter
*
parameter
,
const
std
::
vector
<
lite
::
tensor
::
Tensor
*>
&
in_tensors
,
LiteKernel
(
OpParameter
*
parameter
,
const
std
::
vector
<
lite
::
tensor
::
Tensor
*>
&
in_tensors
,
const
std
::
vector
<
lite
::
tensor
::
Tensor
*>
&
out_tensors
,
const
lite
::
Context
*
ctx
,
const
std
::
vector
<
lite
::
tensor
::
Tensor
*>
&
out_tensors
,
const
lite
::
Context
*
ctx
,
const
mindspore
::
lite
::
PrimitiveC
*
primitive
)
const
mindspore
::
lite
::
PrimitiveC
*
primitive
)
...
@@ -72,12 +73,7 @@ class LiteKernel {
...
@@ -72,12 +73,7 @@ class LiteKernel {
this
->
out_kernels_
.
clear
();
this
->
out_kernels_
.
clear
();
}
}
virtual
~
LiteKernel
()
{
virtual
~
LiteKernel
()
=
default
;
if
(
op_parameter_
!=
nullptr
)
{
delete
op_parameter_
;
op_parameter_
=
nullptr
;
}
}
virtual
int
Prepare
()
{
virtual
int
Prepare
()
{
if
(
!
InferShapeDone
())
{
if
(
!
InferShapeDone
())
{
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录