提交 e46292a9 编写于 作者: B baolei.an

* pass all check hooks

* test=develop
上级 10db87ed
...@@ -101,7 +101,8 @@ int SubgraphEngine::BuildDeviceProgram() { ...@@ -101,7 +101,8 @@ int SubgraphEngine::BuildDeviceProgram() {
static_cast<bm_device_mem_t*>(malloc(sizeof(bm_device_mem_t))); static_cast<bm_device_mem_t*>(malloc(sizeof(bm_device_mem_t)));
CHECK(p_mem != nullptr); CHECK(p_mem != nullptr);
CHECK_EQ(bm_malloc_device_byte(bm_hd_, 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, bmrt_tensor_with_device(&device_outputs_[i], *p_mem,
net_info_->output_dtypes[i], net_info_->output_dtypes[i],
stage.output_shapes[i]); stage.output_shapes[i]);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册