Browse Source

fix opencl mem leak

tags/v1.0.0
wandongdong 5 years ago
parent
commit
805e985e96
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      mindspore/lite/src/runtime/kernel/opencl/subgraph_opencl_kernel.cc

+ 2
- 1
mindspore/lite/src/runtime/kernel/opencl/subgraph_opencl_kernel.cc View File

@@ -103,7 +103,8 @@ int SubGraphOpenCLKernel::GenToFormatOp(const std::vector<lite::Tensor *> &in_te
desc.data_type = kNumberTypeFloat16; desc.data_type = kNumberTypeFloat16;
new_tensor->set_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); MS_ASSERT(parameter);
if (parameter == nullptr) { if (parameter == nullptr) {
MS_LOG(ERROR) << "SubGraphOpenCLKernel new parameter failed!"; MS_LOG(ERROR) << "SubGraphOpenCLKernel new parameter failed!";


Loading…
Cancel
Save