vectorize_scheduler.h

Go to the documentation of this file.

#pragma once
 
#include "sdfg/passes/scheduler/loop_scheduler.h"
 
namespace sdfg {
namespace passes {
namespace scheduler {
 
class VectorizeScheduler : public LoopScheduler {
public:
    SchedulerAction find(
        builder::StructuredSDFGBuilder& builder,
        analysis::AnalysisManager& analysis_manager,
        structured_control_flow::StructuredLoop& loop,
        bool offload_unknown_sizes = false
    ) override;
 
    SchedulerAction find(
        builder::StructuredSDFGBuilder& builder,
        analysis::AnalysisManager& analysis_manager,
        structured_control_flow::While& loop,
        bool offload_unknown_sizes = false
    ) override;
 
    bool can_apply_schedule(
        builder::StructuredSDFGBuilder& builder,
        analysis::AnalysisManager& analysis_manager,
        structured_control_flow::StructuredLoop& loop,
        bool offload_unknown_sizes = false
    ) override;
 
    void apply_schedule(
        builder::StructuredSDFGBuilder& builder,
        analysis::AnalysisManager& analysis_manager,
        structured_control_flow::StructuredLoop& loop,
        bool offload_unknown_sizes = false
    ) override;
 
    static std::string target() {
        return "vectorize";
    };
 
    std::unordered_set<ScheduleTypeCategory> compatible_types() override;
};
 
} // namespace scheduler
} // namespace passes
} // namespace sdfg