tiled_copy.h

Go to the documentation of this file.

#pragma once
 
#include <cstddef>
#include <vector>
 
#include "sdfg/symbolic/assumptions.h"
#include "sdfg/tiles/layout.h"
#include "sdfg/tiles/swizzle.h"
 
namespace sdfg {
namespace tiles {


enum class CopyAtom {
    ScalarSync, 
    VectorSync, 
    CpAsync, 
    CoalesceSync,
};

symbolic::MultiExpression delinearize_rowmajor(const symbolic::Expression& flat, const symbolic::MultiExpression& sizes);

enum class CopyDirection {
    In, 
    Out, 
};

enum class CopyPhase {
    Full, 
    LoadRegs, 
    StoreRegs, 
};
 
struct TiledCopy {
    Layout src{symbolic::MultiExpression{}}; 
    Layout dst{symbolic::MultiExpression{}}; 
    CopyAtom atom = CopyAtom::ScalarSync;
    Swizzle dst_swizzle; 
};

struct TileGuard {
    struct Dim {
        size_t axis; 
        symbolic::Expression base; 
        symbolic::Expression max; 
    };
    symbolic::MultiExpression tile_sizes; 
    std::vector<Dim> dims; 
 
    bool trivial() const {
        return dims.empty();
    }

    symbolic::Condition predicate(const symbolic::MultiExpression& coords) const;
    symbolic::Condition predicate(const symbolic::Expression& flat) const;
 
    void collect_symbols(symbolic::SymbolSet& set) const;
    void replace_symbols(const symbolic::ExpressionMapping& replacements);

    bool discharge(const symbolic::SymbolSet& parameters, const symbolic::Assumptions& assumptions);

    bool simplify();
};

symbolic::Expression simplify_tile_bound(
    const symbolic::Expression& max, const symbolic::Expression& base, const symbolic::Expression& size
);

bool coalesce_sweep_order(TiledCopy& plan, TileGuard& guard);
 
 
} // namespace tiles
} // namespace sdfg