map_collapse.h

Go to the documentation of this file.

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

class MapCollapse : public Transformation {
    structured_control_flow::Map& loop_;
    size_t count_;
    bool applied_ = false;
 
    structured_control_flow::Map* collapsed_loop_ = nullptr;

    bool check_perfect_nest();

    bool check_imperfect(analysis::AnalysisManager& analysis_manager);

    bool is_collapsible_inner_map(structured_control_flow::Map& map, const symbolic::Symbol& outer_indvar);

    void apply_perfect(builder::StructuredSDFGBuilder& builder, analysis::AnalysisManager& analysis_manager);

    void apply_imperfect(builder::StructuredSDFGBuilder& builder, analysis::AnalysisManager& analysis_manager);
 
public:
    MapCollapse(structured_control_flow::Map& loop, size_t count);

    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 MapCollapse from_json(builder::StructuredSDFGBuilder& builder, const nlohmann::json& j);

    structured_control_flow::Map* collapsed_loop();
};
 
} // namespace transformations
} // namespace sdfg