buffer_layout.h

Go to the documentation of this file.

#pragma once
 
#include "sdfg/symbolic/symbolic.h"
#include "sdfg/tiles/layout.h"
#include "sdfg/tiles/swizzle.h"
 
namespace sdfg {
namespace tiles {

enum class Placement {
    RowMajor, 
    ColMajor, 
    Flat, 
};
 
enum class ConflictPolicy {
    None,
    PadRow, 
    PadSlot, 
    Swizzle, 
};

struct PackedBuffer {
    symbolic::MultiExpression slot_sizes; 
    symbolic::MultiExpression tile_sizes; 
    Placement placement = Placement::RowMajor;
    ConflictPolicy conflict = ConflictPolicy::None;
    size_t pad = 0; 

    symbolic::Expression total_size() const;
    symbolic::Expression tile_total_size() const;
    symbolic::Expression slot_stride() const;

    symbolic::MultiExpression axes() const;
    symbolic::MultiExpression
    subset(const symbolic::MultiExpression& slot_indices, const symbolic::MultiExpression& tile_indices) const;
    ComposedLayout layout() const;
};
 
} // namespace tiles
} // namespace sdfg