multi_level_tiling.h

Go to the documentation of this file.

#pragma once
 
#include <cstddef>
#include <vector>
 
#include "sdfg/structured_control_flow/structured_loop.h"
#include "sdfg/transformations/loop_tiling.h"
 
namespace sdfg {
namespace transformations {

class MultiLevelTiling : public LoopTiling {
    std::vector<size_t> additional_tile_sizes_;
 
    std::vector<structured_control_flow::StructuredLoop*> middle_loops_;
 
public:
    MultiLevelTiling(
        structured_control_flow::StructuredLoop& loop, size_t tile_size, size_t tile_size_2, bool simplify_bounds = true
    );

    MultiLevelTiling(
        structured_control_flow::StructuredLoop& loop,
        const std::vector<size_t>& tile_sizes,
        bool simplify_bounds = true
    );
 
    std::string name() const override;
 
    bool can_be_applied(builder::StructuredSDFGBuilder& builder, analysis::AnalysisManager& analysis_manager) override;
 
    void apply(builder::StructuredSDFGBuilder& builder, analysis::AnalysisManager& analysis_manager) override;
 
    void to_json(nlohmann::json& j) const override;
 
    static MultiLevelTiling from_json(builder::StructuredSDFGBuilder& builder, const nlohmann::json& j);
 
    const std::vector<structured_control_flow::StructuredLoop*>& middle_loops();
 
    structured_control_flow::StructuredLoop* middle_loop(size_t idx);

    [[deprecated("Use middle_loop with index instead")]] structured_control_flow::StructuredLoop* middle_loop();
};
 
} // namespace transformations
} // namespace sdfg