scheduler.h

#include <list>
 
#include "smaug/core/network.h"
#include "smaug/core/workspace.h"
#include "smaug/core/operator.h"
 
namespace smaug {
 
class Scheduler {
   public:
    Scheduler(Network* _network, Workspace* _workspace)
            : network(_network), workspace(_workspace) {}
    virtual ~Scheduler(){};
    Tensor* runNetwork();
 
   protected:
    Tensor* scheduleReady();
 
    void maybeRunOperator(Operator* op);
 
    void updateChildren(Operator* op);
 
    Network* network;
    Workspace* workspace;
 
    std::list<Operator*> readyQueue;
};
 
}  // namespace smaug