1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
#include "Halide.h"
#include <stdio.h>
using namespace Halide;
int main(int argc, char **argv) {
Var x("x"), y("y");
Func one("one"), two("two"), three("three"), output("output");
one(x, y) = x + y;
two(x, y) = one(x, y) + 2;
three(x, y) = one(x, y) + 3;
output(x, y) = two(x, y) + three(x, y);
two.compute_root();
one.in(three).compute_root().compute_with(two, Var::outermost());
one.compute_root();
one.compute_at(two, Var::outermost());
output.realize({64, 64});
printf("Success!\n");
return 0;
}
|
