gpu_reduce_layout.h

Go to the documentation of this file.

#pragma once
 
#include <cstdint>
#include <vector>
 
#include "sdfg/symbolic/symbolic.h"
 
namespace sdfg {
namespace gpu {

struct ReductionLayout {
    struct Dimension {
        int64_t stride;
        int64_t count;
    };

    symbolic::Expression base;
    std::vector<Dimension> dimensions;
    int64_t extent = 1;

    ReductionLayout(symbolic::Expression origin, std::vector<Dimension> axes);

    symbolic::Expression unpack(symbolic::Expression slot) const;

    symbolic::Expression pack(symbolic::Expression index) const;
};
 
} // namespace gpu
} // namespace sdfg