重命名 AMEXEC 为 amexec1

2020-06-29 14:42:40 +08:00
parent 4b24915968
commit 1686de8f86
11 changed files with 0 additions and 0 deletions
@@ -0,0 +1,106 @@
+//Model_process.h
+ 
+/**
+* @file model_process.h
+*
+* Copyright (C) 2020. Huawei Technologies Co., Ltd. All rights reserved.
+*
+* This program is distributed in the hope that it will be useful,
+* but WITHOUT ANY WARRANTY; without even the implied warranty of
+* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
+*/
+#pragma once
+#include "utils.h"
+#include "acl/acl.h"
+ 
+using namespace std;
+/**
+* ModelProcess
+*/
+class ModelProcess {
+public:
+    /**
+    * @brief Constructor
+    */
+    ModelProcess();
+ 
+    /**
+    * @brief Destructor
+    */
+    ~ModelProcess();
+ 
+    /**
+    * @brief load model from file with mem
+    * @param [in] modelPath: model path
+    * @return result
+    */
+    Result LoadModelFromFileWithMem(const char *modelPath);
+ 
+    /**
+    * @brief unload model
+    */
+    void Unload();
+ 
+    /**
+    * @brief create model desc
+    * @return result
+    */
+    Result CreateDesc();
+ 
+    /**
+    * @brief destroy desc
+    */
+    void DestroyDesc();
+ 
+    /**
+    * @brief create model input
+    * @param [in] inputDataBuffer: input buffer
+    * @param [in] bufferSize: input buffer size
+    * @return result
+    */
+    Result CreateInput(void *inputDataBuffer, size_t bufferSize);
+ 
+    /**
+    * @brief destroy input resource
+    */
+    void DestroyInput();
+ 
+    /**
+    * @brief create output buffer
+    * @return result
+    */
+    Result CreateOutput();
+ 
+    /**
+    * @brief destroy output resource
+    */
+    void DestroyOutput();
+ 
+    /**
+    * @brief model execute
+    * @return result
+    */
+    Result Execute();
+ 
+    /**
+    * @brief dump model output result to file
+    */
+    void DumpModelOutputResult();
+ 
+    /**
+    * @brief get model output result
+    */
+    void OutputModelResult(std::string& s,std::string& modelName,size_t index);
+	
+ 
+private:
+    uint32_t modelId_;
+    size_t modelMemSize_;
+    size_t modelWeightSize_;
+    void *modelMemPtr_;
+    void *modelWeightPtr_;
+    bool loadFlag_; // model load flag
+    aclmdlDesc *modelDesc_;
+    aclmdlDataset *input_;
+    aclmdlDataset *output_;
+};