From 8c56dadc80416084f959ea47168226168203f0eb Mon Sep 17 00:00:00 2001 From: ling Date: Sat, 17 Apr 2021 16:51:16 +0800 Subject: [PATCH] kernel registry bug --- mindspore/lite/src/kernel_registry.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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; }