fuse_fma.h

Go to the documentation of this file.

#pragma once
 
#include <nlohmann/json_fwd.hpp>
#include <string>
 
#include "sdfg/analysis/analysis.h"
#include "sdfg/builder/structured_sdfg_builder.h"
#include "sdfg/data_flow/access_node.h"
#include "sdfg/data_flow/memlet.h"
#include "sdfg/data_flow/tasklet.h"
#include "sdfg/transformations/transformation.h"
 
namespace sdfg {
namespace fp_math {

class FuseFMA : public transformations::Transformation {
private:
    data_flow::Tasklet& add_;
    data_flow::Tasklet* fma_ = nullptr;
    bool invalidate_analysis_;
 
    struct Match {
        data_flow::Tasklet* multiply = nullptr;
        data_flow::AccessNode* product = nullptr;
        data_flow::Memlet* product_edge = nullptr;
        data_flow::Memlet* addend_edge = nullptr;
    };
 
    bool match(builder::StructuredSDFGBuilder& builder, analysis::AnalysisManager& analysis_manager, Match& out);
 
public:
    explicit FuseFMA(data_flow::Tasklet& add, bool invalidate_analysis = true);
 
    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 FuseFMA from_json(builder::StructuredSDFGBuilder& builder, const nlohmann::json& j);

    data_flow::Tasklet* fma() const;
};
 
} // namespace fp_math
} // namespace sdfg