tile_target.h

Go to the documentation of this file.

#pragma once
 
#include <optional>
 
#include "sdfg/data_flow/library_node.h"
#include "sdfg/structured_control_flow/structured_loop.h"
#include "sdfg/tiles/bank_conflicts.h"
#include "sdfg/tiles/tile.h"
#include "sdfg/types/type.h"
 
namespace sdfg {
namespace tiles {

class TileTarget {
public:
    virtual ~TileTarget() = default;

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

    virtual Space space(Level level) const = 0;

    bool has_scratchpad() const {
        return space(Level::Group) != Space::Global || space(Level::Subgroup) != Space::Global;
    }

    virtual bool supports_cooperative_staging(const structured_control_flow::ScheduleType& sched) const = 0;

    virtual types::StorageType storage_type(Space space) const = 0;

    virtual unsigned lane_width() const = 0;

    virtual std::optional<BankModel> bank_model(Space space) const {
        return std::nullopt;
    }

    virtual bool supports_async_staging(const types::StorageType& storage) const {
        return has_scratchpad() && storage == storage_type(Space::Shared);
    }

    virtual size_t min_async_bytes() const {
        return 4;
    }

    virtual size_t max_tile_elements() const {
        return size_t{1} << 16;
    }

    virtual data_flow::ImplementationType implementation_type() const = 0;
};
 
} // namespace tiles
} // namespace sdfg