From e46292a96ad17be569aa28df68516cdd5ee9dada Mon Sep 17 00:00:00 2001 From: "baolei.an" Date: Sat, 11 Jan 2020 11:27:33 +0800 Subject: [PATCH] * pass all check hooks * test=develop --- lite/kernels/bm/subgraph_compute.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/lite/kernels/bm/subgraph_compute.cc b/lite/kernels/bm/subgraph_compute.cc index a3f11ac78f..16ca8a7e1e 100644 --- a/lite/kernels/bm/subgraph_compute.cc +++ b/lite/kernels/bm/subgraph_compute.cc @@ -101,7 +101,8 @@ int SubgraphEngine::BuildDeviceProgram() { static_cast(malloc(sizeof(bm_device_mem_t))); CHECK(p_mem != nullptr); CHECK_EQ(bm_malloc_device_byte(bm_hd_, - p_mem, origin_otensors_[mapping_index]->memory_size()), BM_SUCCESS); + p_mem, origin_otensors_[mapping_index]->memory_size()), + BM_SUCCESS); bmrt_tensor_with_device(&device_outputs_[i], *p_mem, net_info_->output_dtypes[i], stage.output_shapes[i]); -- GitLab