yankf21
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!7979 fix memory leak 

Merge pull request !7979 from zhaozhenlong/lite/issue/arithmetic_parameter_free_if_needed
This commit is contained in:
mindspore-ci-bot 2020-10-30 09:42:54 +08:00 committed by Gitee
parent 2458c9b322 afafd055e3
commit 460a708ede
2 changed files with 20 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
mindspore/lite/src/runtime/kernel/arm/fp16/arithmetic_fp16.cc
View File

@ -107,7 +107,16 @@ int ArithmeticFP16CPUKernel::PreProcess() {
MS_LOG(ERROR) << "InferShape fail!";
return ret;
}
param_ = reinterpret_cast<ArithmeticParameter *>(PopulateArithmetic(primitive_));
if (op_parameter_ != nullptr) {
free(op_parameter_);
op_parameter_ = nullptr;
}
op_parameter_ = PopulateArithmetic(primitive_);
if (op_parameter_ == nullptr) {
MS_LOG(ERROR) << "Malloc parameter failed";
return RET_ERROR;
}
param_ = reinterpret_cast<ArithmeticParameter *>(op_parameter_);
ret = ReSize();
if (ret != 0) {
MS_LOG(ERROR) << "ReSize fail!ret: " << ret;

11
mindspore/lite/src/runtime/kernel/arm/fp32/arithmetic.cc
View File

@ -48,7 +48,16 @@ int ArithmeticCPUKernel::PreProcess() {
MS_LOG(ERROR) << "InferShape fail!";
return ret;
}
arithmeticParameter_ = reinterpret_cast<ArithmeticParameter *>(PopulateArithmetic(primitive_));
if (op_parameter_ != nullptr) {
free(op_parameter_);
op_parameter_ = nullptr;
}
op_parameter_ = PopulateArithmetic(primitive_);
if (op_parameter_ == nullptr) {
MS_LOG(ERROR) << "Malloc parameter failed";
return RET_ERROR;
}
arithmeticParameter_ = reinterpret_cast<ArithmeticParameter *>(op_parameter_);
ret = ReSize();
if (ret != 0) {
MS_LOG(ERROR) << "ReSize fail!ret: " << ret;