loop_tiling.h

Go to the documentation of this file.

#pragma once
 
#include "sdfg/transformations/transformation.h"
 
namespace sdfg {
namespace transformations {

class LoopTiling : public Transformation {
protected:
    structured_control_flow::StructuredLoop& loop_;
    size_t tile_size_;
    bool simplify_bounds_ = true;
    bool applied_ = false;
 
    structured_control_flow::StructuredLoop* inner_loop_ = nullptr;
    structured_control_flow::StructuredLoop* outer_loop_ = nullptr;

    bool reduction_buffers_supported(
        builder::StructuredSDFGBuilder& builder,
        analysis::AnalysisManager& analysis_manager,
        const std::vector<size_t>& tile_sizes
    ) const;

    static structured_control_flow::StructuredLoop& tile_loop(
        builder::StructuredSDFGBuilder& builder,
        structured_control_flow::StructuredLoop& loop,
        size_t tile_size,
        bool simplify_bounds = true
    );
 
public:
    LoopTiling(structured_control_flow::StructuredLoop& loop, size_t tile_size, bool simplify_bounds = true);

    virtual std::string name() const override;

    virtual bool
    can_be_applied(builder::StructuredSDFGBuilder& builder, analysis::AnalysisManager& analysis_manager) override;

    virtual void apply(builder::StructuredSDFGBuilder& builder, analysis::AnalysisManager& analysis_manager) override;

    virtual void to_json(nlohmann::json& j) const override;

    static LoopTiling from_json(builder::StructuredSDFGBuilder& builder, const nlohmann::json& j);
 
    structured_control_flow::StructuredLoop* inner_loop();
    structured_control_flow::StructuredLoop* outer_loop();
};
 
} // namespace transformations
} // namespace sdfg