swizzle.h

Go to the documentation of this file.

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

struct Swizzle {
    int bits = 0; 
    int base = 0; 
    int shift = 0; 
 
    bool is_identity() const {
        return bits == 0;
    }

    symbolic::Expression apply(const symbolic::Expression& offset) const;
 
    bool operator==(const Swizzle& other) const {
        return bits == other.bits && base == other.base && shift == other.shift;
    }
};

struct ComposedLayout {
    Swizzle swizzle;
    Layout layout{symbolic::MultiExpression{}};

    symbolic::Expression apply_coords(const symbolic::MultiExpression& coords) const;
 
    bool is_plain() const {
        return swizzle.is_identity();
    }
};
 
} // namespace tiles
} // namespace sdfg