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
|
#include "HalideBuffer.h"
#include "HalideRuntime.h"
#include <math.h>
#include <stdio.h>
#include "argvcall.h"
using namespace Halide::Runtime;
const int kSize = 32;
void verify(const Buffer<int32_t, 3> &img, float f1, float f2) {
for (int i = 0; i < kSize; i++) {
for (int j = 0; j < kSize; j++) {
for (int c = 0; c < 3; c++) {
int expected = (int32_t)(c * (i > j ? i : j) * f1 / f2);
if (img(i, j, c) != expected) {
printf("img[%d, %d, %d] = %d (expected %d)\n", i, j, c, img(i, j, c), expected);
exit(1);
}
}
}
}
}
int main(int argc, char **argv) {
int result;
Buffer<int32_t, 3> output(kSize, kSize, 3);
// We can, of course, pass whatever values for Param/ImageParam that we like.
result = argvcall(1.2f, 3.4f, output);
if (result != 0) {
fprintf(stderr, "Result: %d\n", result);
exit(1);
}
verify(output, 1.2f, 3.4f);
// verify that calling via the _argv entry point
// also produces the correct result
float arg0 = 1.234f;
float arg1 = 3.456f;
void *args[3] = {&arg0, &arg1, (halide_buffer_t *)output};
result = argvcall_argv(args);
if (result != 0) {
fprintf(stderr, "Result: %d\n", result);
exit(1);
}
verify(output, arg0, arg1);
printf("Success!\n");
return 0;
}
|
