1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
#include "Halide.h"
#include <iostream>
using namespace Halide;
Buffer<double> test(bool with_vectorize) {
ImageParam input(Float(64), 2);
Func output;
Func input_padded = BoundaryConditions::constant_exterior(input, 100);
RDom rk(-1, 3, -1, 3);
Var x("x"), y("y");
output(x, y) = sum(input_padded(x + rk.x, y + rk.y));
if (with_vectorize) {
output.vectorize(y, 4);
}
Buffer<double> img = lambda(x, y, Expr(1.0)).realize({4, 4});
input.set(img);
Buffer<double> result(4, 4);
input
.dim(0)
.set_bounds(0, 4)
.dim(1)
.set_bounds(0, 4);
output.output_buffer()
.dim(0)
.set_bounds(0, 4)
.dim(1)
.set_bounds(0, 4);
output.realize(result);
return result;
}
int main(int argc, char const *argv[]) {
Buffer<double> im1 = test(true);
Buffer<double> im2 = test(false);
for (int y = 0; y < im1.height(); y++) {
for (int x = 0; x < im1.width(); x++) {
if (im1(x, y) != im2(x, y)) {
printf("im1(%d, %d) = %f, im2(%d, %d) = %f\n",
x, y, im1(x, y),
x, y, im2(x, y));
return 1;
}
}
}
printf("Success!\n");
return 0;
}
|
