diff --git a/mindspore/lite/src/kernel_registry.cc b/mindspore/lite/src/kernel_registry.cc index 8945b366fb9..25518832f5d 100644 --- a/mindspore/lite/src/kernel_registry.cc +++ b/mindspore/lite/src/kernel_registry.cc @@ -39,11 +39,11 @@ KernelRegistry *KernelRegistry::GetInstance() { std::unique_lock malloc_creator_array(instance.lock_); if (instance.creator_arrays_ == nullptr) { - instance.creator_arrays_ = reinterpret_cast(malloc(array_size_ * sizeof(KernelRegistry))); + instance.creator_arrays_ = reinterpret_cast(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; }