Revision 20d6b0f4dd895650c8d755bf2489180743a10d97 authored by Volodymyr Kysenko on 08 January 2024, 19:04:37 UTC, committed by Volodymyr Kysenko on 08 January 2024, 19:04:37 UTC
sliding_vectors.cpp
#include "Halide.h"
using namespace Halide;
int main(int argc, char **argv) {
Func f, g;
Var x;
f(x) = x;
g(x) = f(x - 1) + f(x + 1);
Var xo, xi, xii;
g.split(x, xo, xi, 1024)
.split(xi, xi, xii, 8)
.vectorize(xii);
f.store_at(g, xo)
.compute_at(g, xi)
.vectorize(x, 8);
g.realize({1024});
return 0;
}
Computing file changes ...
