loop_interchange.h

Go to the documentation of this file.

#pragma once
 
#include "sdfg/structured_control_flow/structured_loop.h"
#include "sdfg/tiles/analysis/reduction_buffer_analysis.h"
#include "sdfg/transformations/loop_header.h"
#include "sdfg/transformations/transformation.h"
 
namespace sdfg {
namespace transformations {

class LoopInterchange : public Transformation {
    structured_control_flow::StructuredLoop& outer_loop_;
    structured_control_flow::StructuredLoop& inner_loop_;
    bool applied_ = false;
    structured_control_flow::StructuredLoop* new_outer_loop_;
    structured_control_flow::StructuredLoop* new_inner_loop_;

    bool reduction_buffers_supported(analysis::AnalysisManager& analysis_manager) const;
 
public:
    LoopInterchange(
        structured_control_flow::StructuredLoop& outer_loop, structured_control_flow::StructuredLoop& inner_loop
    );

    virtual std::string name() const override;

    LoopSwap proposal(analysis::AnalysisManager* analysis_manager = nullptr) const;

    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 LoopInterchange from_json(builder::StructuredSDFGBuilder& builder, const nlohmann::json& j);
 
    structured_control_flow::StructuredLoop* new_outer_loop() const;
    structured_control_flow::StructuredLoop* new_inner_loop() const;
};
 
} // namespace transformations
} // namespace sdfg