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

Merge pull request !6694 from chenzupeng/r1.0
pull/6694/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit 22cd1ebd33

@ -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