Enable rtLabelCreateExV2

pull/1043/head
zhangxiaokun 4 years ago
parent 49a67c1177
commit 6ca71007c0

@ -90,7 +90,7 @@ Status RunContextUtil::CreateRtModelResources(uint32_t stream_num, uint32_t even
// Create rt label
for (uint32_t i = 0; i < label_num; ++i) {
rtLabel_t label = nullptr;
rt_ret = rtLabelCreate(&label);
rt_ret = rtLabelCreateV2(&label, rt_model_);
if (rt_ret != RT_ERROR_NONE) {
GELOGE(RT_FAILED, "rtLabelCreate failed. rt_ret = %d, index = %u", static_cast<int>(rt_ret), i);
return RT_FAILED;

@ -1402,7 +1402,7 @@ Status DavinciModel::InitLabelSet(const OpDescPtr &op_desc) {
}
rtLabel_t rt_label = nullptr;
rtError_t rt_error = rtLabelCreateEx(&rt_label, stream);
rtError_t rt_error = rtLabelCreateExV2(&rt_label, rt_model_handle_, stream);
if (rt_error != RT_ERROR_NONE || rt_label == nullptr) {
GELOGE(INTERNAL_ERROR, "InitLabelSet: %s create label failed, error=0x%x.", op_desc->GetName().c_str(), rt_error);
return INTERNAL_ERROR;

Loading…
Cancel
Save