forked from mindspore-Ecosystem/mindspore
!8009 fix memory leak in kernel_registry
Merge pull request !8009 from hangq/primitive
This commit is contained in:
commit
7f58cd214d
|
@ -120,6 +120,8 @@ kernel::LiteKernel *KernelRegistry::GetKernel(const std::vector<Tensor *> &in_te
|
|||
kernel->set_desc(key);
|
||||
}
|
||||
return kernel;
|
||||
} else {
|
||||
free(parameter);
|
||||
}
|
||||
return nullptr;
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue