einsum_passes.h

Go to the documentation of this file.

#pragma once
 
#include <string>
#include "sdfg/analysis/analysis.h"
#include "sdfg/builder/structured_sdfg_builder.h"
#include "sdfg/data_flow/library_node.h"
#include "sdfg/optimization_report/pass_report_consumer.h"
#include "sdfg/passes/pass.h"
#include "sdfg/structured_control_flow/block.h"
#include "sdfg/visitor/structured_sdfg_visitor.h"
 
namespace sdfg {
namespace einsum {
 
class EinsumDetectionPass : public passes::Pass {
public:
    virtual std::string name() override {
        return "EinsumDetectionPass";
    }
 
    virtual bool run_pass(builder::StructuredSDFGBuilder& builder, analysis::AnalysisManager& analysis_manager) override;
};
 
class EinsumConversion : public visitor::NonStoppingStructuredSDFGVisitor {
private:
    PassReportConsumer* report_;
 
public:
    EinsumConversion(
        builder::StructuredSDFGBuilder& builder,
        analysis::AnalysisManager& analysis_manager,
        PassReportConsumer* report = nullptr
    );
 
    static std::string name() {
        return "EinsumConversion";
    }
 
    virtual bool accept(structured_control_flow::Block& block) override;
};
 
typedef passes::VisitorPass<EinsumConversion> EinsumConversionPass;
 
class EinsumLower : public visitor::StructuredSDFGVisitor {
public:
    EinsumLower(builder::StructuredSDFGBuilder& builder, analysis::AnalysisManager& analysis_manager);
 
    static std::string name() {
        return "EinsumLower";
    }
 
    virtual bool accept(structured_control_flow::Block& block) override;
};
 
typedef passes::VisitorPass<EinsumLower> EinsumLowerPass;

class EinsumExpansion : public visitor::StructuredSDFGVisitor {
public:
    EinsumExpansion(builder::StructuredSDFGBuilder& builder, analysis::AnalysisManager& analysis_manager);

    static std::string name() {
        return "EinsumExpansion";
    };

    bool accept(structured_control_flow::Block& node) override;
};

typedef passes::VisitorPass<EinsumExpansion> EinsumExpansionPass;
 
} // namespace einsum
} // namespace sdfg