From bbfe50c7654ae895daef20b721da769857cfe1a4 Mon Sep 17 00:00:00 2001 From: chenzupeng Date: Tue, 22 Sep 2020 09:40:44 +0800 Subject: [PATCH] fix bug: run model shutdown --- .../lite/src/runtime/kernel/opencl/subgraph_opencl_kernel.cc | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/mindspore/lite/src/runtime/kernel/opencl/subgraph_opencl_kernel.cc b/mindspore/lite/src/runtime/kernel/opencl/subgraph_opencl_kernel.cc index 0c4c760aa7..c5d6857fcc 100644 --- a/mindspore/lite/src/runtime/kernel/opencl/subgraph_opencl_kernel.cc +++ b/mindspore/lite/src/runtime/kernel/opencl/subgraph_opencl_kernel.cc @@ -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()); }