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
64
65
66
67
|
#include <cassert>
#include <cstdio>
#include <cstdlib>
#include "HalideBuffer.h"
#include "pipeline_c.h"
#include "pipeline_native.h"
using namespace Halide::Runtime;
extern "C" int an_extern_func(int x, int y) {
return x + y;
}
extern "C" int an_extern_stage(halide_buffer_t *in, halide_buffer_t *out) {
if (in->is_bounds_query()) {
// We expect a 2D input.
in->dim[0].extent = 10;
in->dim[1].extent = 10;
in->dim[0].min = 0;
in->dim[1].min = 0;
} else {
assert(out->host);
int result = 0;
int16_t *origin = (int16_t *)in->host;
origin -= in->dim[0].min * in->dim[0].stride;
origin -= in->dim[1].min * in->dim[1].stride;
for (int y = 0; y < 10; y++) {
for (int x = 0; x < 10; x++) {
result += origin[x * in->dim[0].stride + y * in->dim[1].stride];
}
}
int16_t *dst = (int16_t *)(out->host);
dst[0] = result;
}
return 0;
}
int main(int argc, char **argv) {
Buffer<uint16_t, 2> in(1432, 324);
for (int y = 0; y < in.height(); y++) {
for (int x = 0; x < in.width(); x++) {
in(x, y) = (uint16_t)rand();
}
}
Buffer<uint16_t, 2> out_native(423, 633);
Buffer<uint16_t, 2> out_c(423, 633);
pipeline_native(in, out_native);
pipeline_c(in, out_c);
for (int y = 0; y < out_native.height(); y++) {
for (int x = 0; x < out_native.width(); x++) {
if (out_native(x, y) != out_c(x, y)) {
printf("out_native(%d, %d) = %d, but out_c(%d, %d) = %d\n",
x, y, out_native(x, y),
x, y, out_c(x, y));
}
}
}
printf("Success!\n");
return 0;
}
|
