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(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;