tile_fusion.h

Go to the documentation of this file.

#pragma once
 
#include "sdfg/data_flow/library_nodes/math/tensor/tensor_layout.h"
#include "sdfg/structured_control_flow/map.h"
#include "sdfg/structured_control_flow/sequence.h"
#include "sdfg/structured_control_flow/structured_loop.h"
#include "sdfg/symbolic/assumptions.h"
#include "sdfg/transformations/transformation.h"
 
namespace sdfg {
namespace transformations {

class TileFusion : public Transformation {
    structured_control_flow::Map& first_map_;
    structured_control_flow::Map& second_map_;
    bool applied_ = false;
 
    structured_control_flow::StructuredLoop* fused_loop_ = nullptr;
    int radius_ = 0;

    struct TileFusionCandidate {
        std::string container;
        int radius;
    };
    std::vector<TileFusionCandidate> candidates_;

    struct CyclicContainerInfo {
        std::string container;
        int min_offset; 
        int max_offset; 
        int tiled_dim; 
        int tiled_dim_buf_size; 

        math::tensor::TensorLayout layout{symbolic::MultiExpression{}};
 
        int buffer_size; 
    };
    std::vector<CyclicContainerInfo> cyclic_containers_;

    static int compute_radius(
        const data_flow::Subset& producer_write_subset,
        const std::vector<data_flow::Subset>& consumer_read_subsets,
        const std::vector<structured_control_flow::StructuredLoop*>& producer_loops,
        const std::vector<structured_control_flow::StructuredLoop*>& consumer_loops,
        const symbolic::Assumptions& producer_assumptions,
        const symbolic::Assumptions& consumer_assumptions
    );
 
public:
    TileFusion(structured_control_flow::Map& first_map, structured_control_flow::Map& second_map);
 
    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 TileFusion from_json(builder::StructuredSDFGBuilder& builder, const nlohmann::json& j);

    structured_control_flow::StructuredLoop* fused_loop() const;

    int radius() const;
};
 
} // namespace transformations
} // namespace sdfg