redundant_zero_init_elimination.h

Go to the documentation of this file.

#pragma once
 
#include <string>
#include <vector>
 
#include "sdfg/structured_control_flow/block.h"
#include "sdfg/structured_control_flow/structured_loop.h"
#include "sdfg/transformations/transformation.h"
#include "sdfg/visitor/immutable_structured_sdfg_visitor.h"
 
namespace sdfg {
namespace gpu {
class GpuMmaFragmentLoadNode;
} // namespace gpu
 
namespace transformations {

class MmaNestOutputUses : public visitor::ImmutableStructuredSDFGVisitor {
public:
    struct CFragmentLoad {
        structured_control_flow::Block* block;
        gpu::GpuMmaFragmentLoadNode* node;
    };
 
private:
    structured_control_flow::StructuredLoop& nest_;
    std::string container_;
 
    bool valid_ = true;
    bool has_mma_ = false;
    bool has_store_ = false;
    std::vector<CFragmentLoad> c_loads_;

    bool only_maps_or_branches_enclose(const structured_control_flow::Block& block) const;
 
public:
    MmaNestOutputUses(
        StructuredSDFG& sdfg,
        analysis::AnalysisManager& analysis_manager,
        structured_control_flow::StructuredLoop& nest,
        const std::string& container
    );
 
    bool visit() override;
 
    bool accept(structured_control_flow::Block& node) override;

    bool valid() const {
        return valid_;
    }
    bool has_mma() const {
        return has_mma_;
    }
    bool has_store() const {
        return has_store_;
    }
    const std::vector<CFragmentLoad>& c_loads() const {
        return c_loads_;
    }
};

class RedundantZeroInitElimination : public Transformation {
    structured_control_flow::StructuredLoop& loop_;
 
public:
    explicit RedundantZeroInitElimination(structured_control_flow::StructuredLoop& loop);
 
    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 RedundantZeroInitElimination from_json(builder::StructuredSDFGBuilder& builder, const nlohmann::json& j);
};
 
} // namespace transformations
} // namespace sdfg