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
|
#include "Halide.h"
#include <stdio.h>
using namespace Halide;
int main(int argc, char **argv) {
if (get_jit_target_from_environment().arch == Target::WebAssembly) {
printf("[SKIP] WebAssembly JIT does not support Param<> for pointer types.\n");
return 0;
}
Func f, g;
Var x;
Param<void *> handle;
f(x) = reinterpret<uint64_t>(handle);
g(x) = reinterpret<uint64_t>(handle);
int foo;
handle.set(&foo);
Buffer<uint64_t> out1 = f.realize({4});
g.vectorize(x, 4);
Buffer<uint64_t> out2 = g.realize({4});
uint64_t correct = (uint64_t)((uintptr_t)(&foo));
for (int x = 0; x < out1.width(); x++) {
if (out1(x) != correct) {
printf("out1(%d) = %llu instead of %llu\n",
x,
(long long unsigned)out1(x),
(long long unsigned)correct);
return 1;
}
if (out2(x) != correct) {
printf("out2(%d) = %llu instead of %llu\n",
x,
(long long unsigned)out2(x),
(long long unsigned)correct);
return 1;
}
}
printf("Success!\n");
return 0;
}
|
