From 805e985e966baaf9a6b751ef555f9bbaf103d6a7 Mon Sep 17 00:00:00 2001 From: wandongdong Date: Mon, 21 Sep 2020 02:54:31 -0700 Subject: [PATCH] fix opencl mem leak --- .../lite/src/runtime/kernel/opencl/subgraph_opencl_kernel.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/mindspore/lite/src/runtime/kernel/opencl/subgraph_opencl_kernel.cc b/mindspore/lite/src/runtime/kernel/opencl/subgraph_opencl_kernel.cc index d14bdddce6..0c4c760aa7 100644 --- a/mindspore/lite/src/runtime/kernel/opencl/subgraph_opencl_kernel.cc +++ b/mindspore/lite/src/runtime/kernel/opencl/subgraph_opencl_kernel.cc @@ -103,7 +103,8 @@ int SubGraphOpenCLKernel::GenToFormatOp(const std::vector &in_te desc.data_type = kNumberTypeFloat16; new_tensor->set_data_type(kNumberTypeFloat16); } - OpenCLToFormatParameter *parameter = new (std::nothrow) OpenCLToFormatParameter; + OpenCLToFormatParameter *parameter = + static_cast(malloc(sizeof(OpenCLToFormatParameter))); MS_ASSERT(parameter); if (parameter == nullptr) { MS_LOG(ERROR) << "SubGraphOpenCLKernel new parameter failed!";