ComputeAction.h

 
#ifndef WRENCH_COMPUTEACTION_H
#define WRENCH_COMPUTEACTION_H
 
#include <string>
 
#include "wrench/action/Action.h"
 
namespace wrench {
 
    class ComputeThread;
    class ParallelModel;
 
    class ComputeAction : public Action {
 
    public:
        double getFlops() const;
        unsigned long getMinNumCores() const override;
        unsigned long getMaxNumCores() const override;
        double getMinRAMFootprint() const override;
        std::shared_ptr<ParallelModel> getParallelModel() const;
 
    protected:
        friend class CompoundJob;
 
        ComputeAction(std::string name,
                      std::shared_ptr<CompoundJob> job,
                      double flops,
                      double ram,
                      unsigned long min_core,
                      unsigned long max_core,
                      std::shared_ptr<ParallelModel> parallel_model);
 
        void execute(std::shared_ptr<ActionExecutor> action_executor) override;
        void terminate(std::shared_ptr<ActionExecutor> action_executor) override;
 
    private:
        double flops;
        unsigned long min_num_cores;
        unsigned long max_num_cores;
        double ram;
        std::shared_ptr<ParallelModel> parallel_model;
 
        void simulateComputationAsSleep(std::shared_ptr<ActionExecutor> action_executor, std::vector<double> &work_per_thread);
        void simulateComputationWithComputeThreads(std::shared_ptr<ActionExecutor> action_executor, std::vector<double> &work_per_thread);
 
        std::vector<std::shared_ptr<ComputeThread>> compute_threads;
 
 
    };
}
 
 
#endif //WRENCH_COMPUTEACTION_H