From 5f6f7f23d8b7d8d0e248dde4e5dce9700b58ba97 Mon Sep 17 00:00:00 2001
From: zhangxiaokun <zhang.xiaokun@huawei.com>
Date: Fri, 29 Jan 2021 09:11:11 +0800
Subject: [PATCH] Add rtLabelCreateV2 stub

---
 tests/depends/runtime/src/runtime_stub.cc | 32 +++++++++++++++++------
 1 file changed, 24 insertions(+), 8 deletions(-)

diff --git a/tests/depends/runtime/src/runtime_stub.cc b/tests/depends/runtime/src/runtime_stub.cc
index 9b45e7e2..bd256aa9 100644
--- a/tests/depends/runtime/src/runtime_stub.cc
+++ b/tests/depends/runtime/src/runtime_stub.cc
@@ -245,9 +245,31 @@ rtError_t rtProfilerInit(const char *prof_dir, const char *address, const char *
 
 rtError_t rtProfilerStart(void) { return RT_ERROR_NONE; }
 
-rtError_t rtLabelCreate(rtLabel_t *label) { return RT_ERROR_NONE; }
+rtError_t rtLabelCreate(rtLabel_t *label) {
+  *label = new uint64_t;
+  return RT_ERROR_NONE;
+}
 
-rtError_t rtLabelDestroy(rtLabel_t label) { return RT_ERROR_NONE; }
+rtError_t rtLabelCreateEx(rtLabel_t *label, rtStream_t stream) {
+  *label = new uint64_t;
+  return RT_ERROR_NONE;
+}
+
+rtError_t rtLabelCreateV2(rtLabel_t *label, rtModel_t model) {
+  *label = new uint64_t;
+  return RT_ERROR_NONE;
+}
+
+rtError_t rtLabelCreateExV2(rtLabel_t *label, rtModel_t model, rtStream_t stream) {
+  *label = new uint64_t;
+  return RT_ERROR_NONE;
+}
+
+rtError_t rtLabelDestroy(rtLabel_t label) {
+  uint64_t *stub = static_cast<uint64_t>(label);
+  delete stub;
+  return RT_ERROR_NONE;
+}
 
 rtError_t rtLabelSet(rtLabel_t label, rtStream_t stream) { return RT_ERROR_NONE; }
 
@@ -364,12 +386,6 @@ rtError_t rtSetCtxINFMode(bool mode)
   return RT_ERROR_NONE;
 }
 
-rtError_t rtLabelCreateEx(rtLabel_t *label, rtStream_t stream)
-{
-  *label = new uint32_t;
-  return RT_ERROR_NONE;
-}
-
 rtError_t rtGetRtCapability(rtFeatureType_t featureType, int32_t featureInfo, int64_t *value)
 {
   return RT_ERROR_NONE;