Browse Source

!6693 [MS][LITE][GPU]fix bug: run model shutdown

Merge pull request !6693 from chenzupeng/master-lite-2
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
93d742732e
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