fix memory leak in kernel_registry · b45e7847c8 - mindspore - 开源协同云脑生态支撑系统

fix memory leak in kernel_registry

tags/v1.1.0

hangangqiang 5 years ago

parent

542a52fbf8

commit

b45e7847c8

1 changed files with 2 additions and 0 deletions

Unified View

							
								+ 2
								
								- 0
							
mindspore/lite/src/kernel_registry.ccView File
								
																@@ -120,6 +120,8 @@ kernel::LiteKernel *KernelRegistry::GetKernel(const std::vector<Tensor *> &in_te
															
																      kernel->set_desc(key);
																
																      kernel->set_desc(key);
															
																    }
																
																    }
															
																    return kernel;
																
																    return kernel;
															
																  } else {
															
																    free(parameter);
															
																  }
																
																  }
															
																  return nullptr;
																
																  return nullptr;
															
																}
																
																}