From 52a7de08d64d0af15f053b0e9cfa0bd2ae3204bf Mon Sep 17 00:00:00 2001 From: yeyunpeng2020 Date: Thu, 1 Apr 2021 10:22:03 +0800 Subject: [PATCH] fix ci bug --- mindspore/lite/src/runtime/kernel/opencl/kernel/resize.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mindspore/lite/src/runtime/kernel/opencl/kernel/resize.cc b/mindspore/lite/src/runtime/kernel/opencl/kernel/resize.cc index 4e0e89b840..ecb6bc3bef 100644 --- a/mindspore/lite/src/runtime/kernel/opencl/kernel/resize.cc +++ b/mindspore/lite/src/runtime/kernel/opencl/kernel/resize.cc @@ -33,7 +33,7 @@ using mindspore::schema::PrimitiveType_Resize; namespace mindspore::kernel { int ResizeOpenCLKernel::CheckSpecs() { - if (in_tensors_.size() != 2 || out_tensors_.size() != 1) { + if (!(in_tensors_.size() == 1 || in_tensors_.size() == 2) || out_tensors_.size() != 1) { MS_LOG(ERROR) << "in size: " << in_tensors_.size() << ", out size: " << out_tensors_.size(); return RET_ERROR; } @@ -120,7 +120,7 @@ int ResizeOpenCLKernel::Run() { } int ResizeOpenCLKernel::PreProcess() { - if (Type() == PrimitiveType_Resize && !op_parameter_->infer_flag_) { + if (Type() == PrimitiveType_Resize && !op_parameter_->infer_flag_ && in_tensors_.size() == 2) { auto shape_tensor = in_tensors_[1]; if (!shape_tensor->IsConst()) { ocl_runtime_->SyncCommandQueue();