Description:Support model_exit in GE Team:HISI_SW Feature or Bugfix:Featurepull/131/head
l00444296
4 years ago
parent
3eadab4c6a
commit
9b35666be5
@ -0,0 +1,54 @@
|
||||
/**
|
||||
* Copyright 2019-2020 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "graph/load/new_model_manager/task_info/model_exit_task_info.h"
|
||||
|
||||
#include "common/properties_manager.h"
|
||||
#include "framework/common/debug/ge_log.h"
|
||||
#include "graph/load/new_model_manager/davinci_model.h"
|
||||
|
||||
namespace ge {
|
||||
Status ModelExitTaskInfo::Init(const domi::TaskDef &task_def, DavinciModel *davinci_model) {
|
||||
GELOGI("InitModelExitTaskInfo Init Start.");
|
||||
if (davinci_model == nullptr) {
|
||||
GELOGE(PARAM_INVALID, "davinci_model is null!");
|
||||
return PARAM_INVALID;
|
||||
}
|
||||
|
||||
Status ret = SetStream(task_def.stream_id(), davinci_model->GetStreamList());
|
||||
if (ret != SUCCESS) {
|
||||
GELOGE(ret, "SetStream fail, stream_id:%u", task_def.stream_id());
|
||||
return ret;
|
||||
}
|
||||
|
||||
model_ = davinci_model->GetRtModelHandle();
|
||||
GELOGI("InitModelExitTaskInfo Init Success, model:%p, stream:%p", model_, stream_);
|
||||
return SUCCESS;
|
||||
}
|
||||
|
||||
Status ModelExitTaskInfo::Distribute() {
|
||||
GELOGI("ModelExitTaskInfo Distribute Start.");
|
||||
rtError_t rt_ret = rtModelExit(model_, stream_);
|
||||
if (rt_ret != RT_ERROR_NONE) {
|
||||
GELOGE(RT_FAILED, "Call rtModelExit failed, ret: 0x%x", rt_ret);
|
||||
return RT_ERROR_TO_GE_STATUS(rt_ret);
|
||||
}
|
||||
GELOGI("ModelExitTaskInfo Distribute Success.");
|
||||
return SUCCESS;
|
||||
}
|
||||
|
||||
REGISTER_TASK_INFO(RT_MODEL_TASK_MODEL_EXIT, ModelExitTaskInfo);
|
||||
} // namespace ge
|
||||
@ -0,0 +1,37 @@
|
||||
/**
|
||||
* Copyright 2019-2020 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef GE_GRAPH_LOAD_NEW_MODEL_MANAGER_TASK_INFO_MODEL_EXIT_TASK_INFO_H_
|
||||
#define GE_GRAPH_LOAD_NEW_MODEL_MANAGER_TASK_INFO_MODEL_EXIT_TASK_INFO_H_
|
||||
|
||||
#include "graph/load/new_model_manager/task_info/task_info.h"
|
||||
|
||||
namespace ge {
|
||||
class ModelExitTaskInfo : public TaskInfo {
|
||||
public:
|
||||
ModelExitTaskInfo() {}
|
||||
|
||||
~ModelExitTaskInfo() override { model_ = nullptr; }
|
||||
|
||||
Status Init(const domi::TaskDef &task_def, DavinciModel *davinci_model) override;
|
||||
|
||||
Status Distribute() override;
|
||||
|
||||
private:
|
||||
rtModel_t model_{nullptr};
|
||||
};
|
||||
} // namespace ge
|
||||
#endif // GE_GRAPH_LOAD_NEW_MODEL_MANAGER_TASK_INFO_MODEL_EXIT_TASK_INFO_H_
|
||||
Loading…
Reference in new issue