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

kernel registry bug

This commit is contained in:
ling 2021-04-17 16:51:16 +08:00
parent 449c4ba00e
commit 8c56dadc80
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_);
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) {
return nullptr;
}
memset(instance.creator_arrays_, 0, array_size_ * sizeof(KernelRegistry));
memset(instance.creator_arrays_, 0, array_size_ * sizeof(KernelCreator));
}
return &instance;
}