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
|
#include "Halide.h"
#include <stdio.h>
using namespace Halide;
using namespace Halide::Internal;
struct struct_t {
double a;
int32_t b;
int16_t c;
const char *d;
};
extern "C" HALIDE_EXPORT_SYMBOL int check_struct(struct_t *s) {
if (s->a != 3.0 ||
s->b != 1234567 ||
s->c != 1234 ||
strcmp(s->d, "Test global string\n")) {
printf("Unexpected struct values: %f %d %d %s\n", s->a, s->b, s->c, s->d);
exit(1);
}
return 0;
}
HalideExtern_1(int, check_struct, struct_t *);
int main(int argc, char **argv) {
if (get_jit_target_from_environment().arch == Target::WebAssembly) {
printf("[SKIP] Skipping test for WebAssembly as the wasm JIT cannot support passing arbitrary pointers to/from HalideExtern code.\n");
return 0;
}
// Check make_struct is working. make_struct is not intended to be
// called from the front-end because the structs live on the stack
// of the generated function. The generated structs should also
// not be stored in Funcs. They're just pointers to a single stack
// slot. There's also no way to extract fields from the struct
// without an extern function. You can really only use them for
// marshalling some arguments to immediately pass to an extern
// call, and that's what they're used for in the runtime.
Expr a = cast<double>(3.0f);
Expr b = cast<int32_t>(1234567);
Expr c = cast<int16_t>(1234);
Expr d = std::string("Test global string\n");
Expr s = Call::make(Handle(), Call::make_struct, {a, b, c, d}, Call::Intrinsic);
Func g;
g() = check_struct(s);
g.realize();
printf("Success!\n");
return 0;
}
|
