Revision 16ddff55efc02d37b713eed569d435bdc4f5dfb7 authored by Andrew Adams on 31 August 2023, 22:21:03 UTC, committed by Andrew Adams on 31 August 2023, 22:21:03 UTC
1 parent ef9a7d8
cast_handle.cpp
#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;
}
Computing file changes ...