diff --git a/lite/kernels/bm/subgraph_compute.cc b/lite/kernels/bm/subgraph_compute.cc index a3f11ac78f990c2bff74399e2210332a84701d77..16ca8a7e1eaf0e8bc3518359beefcb3341167bfb 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]);