task.h

Go to the documentation of this file.

#ifndef UNI_INCLUDE_TASK_H_
#define UNI_INCLUDE_TASK_H_

#include <string>
#include <map>
#include <memory>
#include <iostream>
#include <sstream>

#include "tensor.hpp"
#include "profiling.h"

typedef enum TaskStatus {
    TASK_CREATE,  
    TASK_READY,   
    TASK_RUN,     
    TASK_END      
} TaskStatus;

class Task {
public:
    Task()
    {
        this->status = TASK_CREATE;
    }

    Task(std::string graphPath, std::map<std::string, std::shared_ptr<Tensor>> data)
    {
        this->set(ut_time_ms(), graphPath, data, TASK_READY);
    }

    Task(int id, std::string graphPath, std::map<std::string, std::shared_ptr<Tensor>> data)
    {
        this->set(id, graphPath, data, TASK_READY);
    }

    Task(Task *task)
    {
        this->set(task->id, task->graphPath, task->data, task->status);
    }

    void set(int id,
        std::string graphPath,
        std::map<std::string, std::shared_ptr<Tensor>> data,
        TaskStatus status)
    {
        this->id = id;
        this->graphPath = graphPath;
        this->data = data;
        this->status = status;
    }

    friend std::ostream &operator<<(std::ostream &os, const Task &task)
    {
        os << "Task " << task.id << "(timestamp " << task.id << ", status " << task.status
           << ", graph " << task.graphPath << ", data " << std::endl;
        for (auto iter : task.data) {
            os << "tensor name " << iter.first << " " << iter.second->string(1) << std::endl;
        }
        os << ")";
        return os;
    }

    int id;
    TaskStatus status;
    std::string graphPath;
    std::map<std::string, std::shared_ptr<Tensor>> data;
};
#endif  // UNI_INCLUDE_TASK_H_