loop_scheduler.h

Go to the documentation of this file.

#pragma once
 
#include <sdfg/analysis/loop_analysis.h>
#include <sdfg/passes/pass.h>
#include <sdfg/transformations/recorder.h>
#include <unordered_set>
#include <vector>
#include "sdfg/optimization_report/pass_report_consumer.h"
#include "sdfg/structured_control_flow/map.h"
 
namespace sdfg {
namespace passes {
namespace scheduler {
 
enum SchedulerAction {
    APPLY,
    NEXT,
    CHILDREN,
};
 
struct SchedulerLoopInfo {
    // Static Properties
    analysis::LoopInfo loop_info = analysis::LoopInfo();
};
 
 
class LoopScheduler {
protected:
    PassReportConsumer* report_ = nullptr;
    transformations::Recorder* recorder_ = nullptr;
 
public:
    virtual ~LoopScheduler() = default;

    virtual SchedulerAction find(
        builder::StructuredSDFGBuilder& builder,
        analysis::AnalysisManager& analysis_manager,
        structured_control_flow::StructuredLoop& loop,
        bool offload_unknown_sizes = false
    ) = 0;
 
    virtual SchedulerAction find(
        builder::StructuredSDFGBuilder& builder,
        analysis::AnalysisManager& analysis_manager,
        structured_control_flow::While& loop,
        bool offload_unknown_sizes = false
    ) = 0;

    virtual void pre_schedule(
        builder::StructuredSDFGBuilder& builder,
        analysis::AnalysisManager& analysis_manager,
        std::vector<structured_control_flow::StructuredLoop*>& applicable_loops
    ) {
    }

    virtual bool can_apply_schedule(
        builder::StructuredSDFGBuilder& builder,
        analysis::AnalysisManager& analysis_manager,
        structured_control_flow::StructuredLoop& loop,
        bool offload_unknown_sizes = false
    ) = 0;

    virtual void apply_schedule(
        builder::StructuredSDFGBuilder& builder,
        analysis::AnalysisManager& analysis_manager,
        structured_control_flow::StructuredLoop& loop,
        bool offload_unknown_sizes = false
    ) = 0;

    virtual bool post_schedule(
        builder::StructuredSDFGBuilder& builder,
        analysis::AnalysisManager& analysis_manager,
        std::vector<structured_control_flow::StructuredLoop*>& scheduled_loops
    ) {
        return false;
    }
 
    virtual void set_report(PassReportConsumer* report) {
        report_ = report;
    }

    virtual void set_recorder(transformations::Recorder* recorder) {
        recorder_ = recorder;
    }
 
    virtual std::unordered_set<ScheduleTypeCategory> compatible_types() = 0;
};
 
} // namespace scheduler
} // namespace passes
} // namespace sdfg