!6657 fix opencl mem leak

Merge pull request !6657 from wandongdong/master
4 years ago · e2821d0769
parent dde9f5ac25 805e985e96
commit e2821d0769
1 changed files with 2 additions and 1 deletions
--- 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<lite::Tensor *> &in_te
      desc.data_type = kNumberTypeFloat16;
      new_tensor->set_data_type(kNumberTypeFloat16);
    }
-    OpenCLToFormatParameter *parameter = new (std::nothrow) OpenCLToFormatParameter;
+    OpenCLToFormatParameter *parameter =
+      static_cast<OpenCLToFormatParameter *>(malloc(sizeof(OpenCLToFormatParameter)));
    MS_ASSERT(parameter);
    if (parameter == nullptr) {
      MS_LOG(ERROR) << "SubGraphOpenCLKernel new parameter failed!";