diff --git a/lite/api/cxx_api.cc b/lite/api/cxx_api.cc index f4dcac519a0699cbcf1bdd3845d8ae90d7a289ed..f123c2a9c7db336b6e94b0c1934fc2e284d50f67 100644 --- a/lite/api/cxx_api.cc +++ b/lite/api/cxx_api.cc @@ -293,6 +293,7 @@ void Predictor::Build(const cpp::ProgramDesc &desc, // `inner_places` is used to optimize passes std::vector inner_places = valid_places; for (auto &valid_place : valid_places) { + if (valid_place.target == TARGET(kOpenCL)) continue; inner_places.emplace_back( Place(TARGET(kHost), valid_place.precision, valid_place.layout)); } diff --git a/lite/kernels/opencl/slice_image_compute_test.cc b/lite/kernels/opencl/slice_image_compute_test.cc index 1c22a81238d5c586602244e6bb1873f3a4582802..b8e8e18af17ef6f104708d4ef0cee4db5f3ff5b6 100644 --- a/lite/kernels/opencl/slice_image_compute_test.cc +++ b/lite/kernels/opencl/slice_image_compute_test.cc @@ -84,7 +84,7 @@ TEST(slice_image2d_fp16, compute) { } LOG(INFO) << "prepare input"; - std::shared_ptr default_converter( + std::unique_ptr default_converter( new CLImageConverterDefault()); DDim image_shape = default_converter->InitImageDimInfoWith(in_dim); LOG(INFO) << "image_shape = " << image_shape[0] << " " << image_shape[1];