Browse Source

fix bug: run model shutdown

tags/v1.1.0
chenzupeng 5 years ago
parent
commit
bbfe50c765
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      mindspore/lite/src/runtime/kernel/opencl/subgraph_opencl_kernel.cc

+ 4
- 0
mindspore/lite/src/runtime/kernel/opencl/subgraph_opencl_kernel.cc View File

@@ -321,6 +321,10 @@ int SubGraphOpenCLKernel::ReSize() { return RET_OK; }

int SubGraphOpenCLKernel::Run() {
for (auto &tensor : in_tensors_) {
if (tensor->data_c() == nullptr) {
MS_LOG(ERROR) << "OpenCL subgraph input tensor data is null";
return RET_ERROR;
}
allocator_->UnmapBuffer(tensor->data_c());
}



Loading…
Cancel
Save