commit tools

AMEXEC/inc/model_process.h

@@ -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_;
+};

AMEXEC/inc/sample_process.h

@@ -0,0 +1,50 @@
+//Sample_process.h
+ 
+/**
+* @file sample_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;
+/**
+* SampleProcess
+*/
+class SampleProcess {
+public:
+    /**
+    * @brief Constructor
+    */
+    SampleProcess();
+ 
+    /**
+    * @brief Destructor
+    */
+    ~SampleProcess();
+ 
+    /**
+    * @brief init reousce
+    * @return result
+    */
+    Result InitResource();
+ 
+    /**
+    * @brief sample process
+    * @return result
+    */
+    Result Process(vector<string>& params, vector<string>& inputs, size_t idx);
+ 
+private:
+    void DestroyResource();
+ 
+    int32_t deviceId_;
+    aclrtContext context_;
+    aclrtStream stream_;
+};

AMEXEC/inc/utils.h

@@ -0,0 +1,71 @@
+//Util.h
+ 
+/**
+* @file utils.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 <iostream>
+#include <sstream>
+#include <fstream>
+#include <cstring>
+#include <string>
+#include <map>
+#include <vector>
+#include <algorithm>
+#include <stdio.h>
+#include <time.h>
+ 
+#define INFO_LOG(fmt, args...) fprintf(stdout, "[INFO] " fmt "\n", ##args)
+#define WARN_LOG(fmt, args...) fprintf(stdout, "[WARN] " fmt "\n", ##args)
+#define ERROR_LOG(fmt, args...) fprintf(stdout, "[ERROR] " fmt "\n", ##args)
+ 
+using namespace std;
+static size_t loop = 1;
+typedef enum Result {
+    SUCCESS = 0,
+    FAILED = 1
+} Result;
+
+ 
+/**
+* Utils
+*/
+class Utils {
+public:
+    /**
+    * @brief create device buffer of file
+    * @param [in] fileName: file name
+    * @param [out] fileSize: size of file
+    * @return device buffer of file
+    */
+    static void *GetDeviceBufferOfFile(std::string fileName, uint32_t &fileSize);
+ 
+    /**
+    * @brief create buffer of file
+    * @param [in] fileName: file name
+    * @param [out] fileSize: size of file
+    * @return buffer of pic
+    */
+    static void* ReadBinFile(std::string fileName, uint32_t& fileSize);
+ 
+    static void SplitString(std::string& s, std::vector<std::string>& v, char c);
+        
+    static int str2num(char *str);
+	
+	static std::string modelName(string& s);
+	
+	static std::string TimeLine();
+	
+	static void printCurrentTime();
+	
+	static void printHelpLetter();	
+	
+};
+ 
+#pragma once