reduce_dispatcher.h

Go to the documentation of this file.

#pragma once
 
#include "sdfg/codegen/code_snippet_factory.h"
#include "sdfg/codegen/dispatchers/node_dispatcher.h"
#include "sdfg/structured_control_flow/reduce.h"
 
namespace sdfg {
namespace codegen {

class SchedTypeReduceDispatcher : public NodeDispatcher {
private:
    structured_control_flow::Reduce& node_;
 
public:
    SchedTypeReduceDispatcher(
        LanguageExtension& language_extension,
        StructuredSDFG& sdfg,
        analysis::AnalysisManager& analysis_manager,
        structured_control_flow::Reduce& node,
        InstrumentationPlan& instrumentation_plan,
        ArgCapturePlan& arg_capture_plan
    );
 
    void dispatch_node(
        PrettyPrinter& main_stream, PrettyPrinter& globals_stream, CodeSnippetFactory& library_snippet_factory
    ) override {
        throw std::runtime_error("ReduceDispatcher::dispatch_node not implemented");
    }
 
    void dispatch(
        PrettyPrinter& main_stream, PrettyPrinter& globals_stream, CodeSnippetFactory& library_snippet_factory
    ) override;
 
    InstrumentationInfo instrumentation_info() const override;
};

typedef SchedTypeReduceDispatcher ReduceDispatcher;

class SequentialReduceDispatcher : public NodeDispatcher {
private:
    structured_control_flow::Reduce& node_;
 
public:
    SequentialReduceDispatcher(
        LanguageExtension& language_extension,
        StructuredSDFG& sdfg,
        analysis::AnalysisManager& analysis_manager,
        structured_control_flow::Reduce& node,
        InstrumentationPlan& instrumentation_plan,
        ArgCapturePlan& arg_capture_plan
    );
 
    void dispatch_node(
        PrettyPrinter& main_stream, PrettyPrinter& globals_stream, CodeSnippetFactory& library_snippet_factory
    ) override;
 
    InstrumentationInfo instrumentation_info() const override;
};
 
} // namespace codegen
} // namespace sdfg