Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!15335 kernel registry bug 

From: @ling_qiao_min
Reviewed-by: @zhang_xue_tong,@zhanghaibo5
Signed-off-by: @zhang_xue_tong
This commit is contained in:
mindspore-ci-bot 2021-04-17 18:12:18 +08:00 committed by Gitee
parent 858924a528 8c56dadc80
commit 9e6f9aa56a
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/lite/src/kernel_registry.cc
View File

@ -39,11 +39,11 @@ KernelRegistry *KernelRegistry::GetInstance() {
std::unique_lock<std::mutex> malloc_creator_array(instance.lock_); std::unique_lock<std::mutex> malloc_creator_array(instance.lock_);
if (instance.creator_arrays_ == nullptr) { if (instance.creator_arrays_ == nullptr) {
instance.creator_arrays_ = reinterpret_cast<KernelCreator *>(malloc(array_size_ * sizeof(KernelRegistry))); instance.creator_arrays_ = reinterpret_cast<KernelCreator *>(malloc(array_size_ * sizeof(KernelCreator)));
if (instance.creator_arrays_ == nullptr) { if (instance.creator_arrays_ == nullptr) {
return nullptr; return nullptr;
} }
memset(instance.creator_arrays_, 0, array_size_ * sizeof(KernelRegistry)); memset(instance.creator_arrays_, 0, array_size_ * sizeof(KernelCreator));
} }
return &instance; return &instance;
} }