offloading_type_collector.h

Go to the documentation of this file.

#pragma once
 
#include <sdfg/data_flow/library_nodes/math/blas/blas_node.h>
#include <sdfg/data_flow/library_nodes/math/math_node.h>
#include <sdfg/structured_control_flow/map.h>
#include <sdfg/visitor/structured_sdfg_visitor.h>
 
namespace sdfg::auto_util {

class OffloadingTypeCollector : public sdfg::visitor::StructuredSDFGVisitor {
private:
    ScheduleType schedule_type_;
 
public:
    OffloadingTypeCollector(
        sdfg::builder::StructuredSDFGBuilder& builder,
        sdfg::analysis::AnalysisManager& analysis_manager,
        const ScheduleType& sched_type
    );
 
    bool accept(sdfg::structured_control_flow::Block& node) override;
 
    bool accept(sdfg::structured_control_flow::Map& map) override;
 
    const ScheduleType& schedule_type() const;
 
private:
    void found_schedule_type(const ScheduleType& sched_type);
};
 
} // namespace sdfg::auto_util