tile_target_registry.h

Go to the documentation of this file.

#pragma once
 
#include <memory>
#include <mutex>
#include <string>
#include <unordered_map>
 
#include "sdfg/tiles/tile_target.h"
 
namespace sdfg {
namespace tiles {

class TileTargetRegistry {
    mutable std::mutex mutex_;
    std::unordered_map<std::string, std::shared_ptr<TileTarget>> targets_;
 
public:
    static TileTargetRegistry& instance() {
        static TileTargetRegistry registry;
        return registry;
    }
 
    void register_target(const std::string& schedule_value, std::shared_ptr<TileTarget> target) {
        std::lock_guard<std::mutex> lock(mutex_);
        targets_.emplace(schedule_value, std::move(target));
    }

    const TileTarget* get(const std::string& schedule_value) const {
        std::lock_guard<std::mutex> lock(mutex_);
        auto it = targets_.find(schedule_value);
        return it == targets_.end() ? nullptr : it->second.get();
    }

    data_flow::ImplementationType implementation_type(const std::string& schedule_value) const {
        const TileTarget* target = get(schedule_value);
        return target ? target->implementation_type() : data_flow::ImplementationType_NONE;
    }

    std::optional<AxisSchedule> classify(const structured_control_flow::ScheduleType& sched) const;

    std::optional<Level> classify_level(const structured_control_flow::ScheduleType& sched) const;

    bool drives_cooperative_copy(const structured_control_flow::ScheduleType& sched) const;
};
 
} // namespace tiles
} // namespace sdfg