p85620714
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

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

Merge pull request !6693 from chenzupeng/master-lite-2
This commit is contained in:
mindspore-ci-bot 2020-09-22 11:41:50 +08:00 committed by Gitee
parent d76ccefcdc bbfe50c765
commit 93d742732e
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
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());
}