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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
#include <stdio.h>
#include "HalideBuffer.h"
#include "HalideRuntime.h"
#include <assert.h>
#include <string.h>
#include <string>
#include "cxx_mangling.function_info.h"
#include "cxx_mangling.h"
#ifdef TEST_CUDA
#include "cxx_mangling_gpu.h"
#endif
using namespace Halide::Runtime;
namespace my_namespace {
class my_class {
public:
int foo;
};
namespace my_subnamespace {
struct my_struct {
int foo;
};
} // namespace my_subnamespace
} // namespace my_namespace
union my_union {
float a;
int b;
};
int main(int argc, char **argv) {
Buffer<uint8_t, 1> input(100);
for (int32_t i = 0; i < 100; i++) {
input(i) = i;
}
Buffer<double, 1> result(100);
const halide_filter_metadata_t *m = HalideTest::AnotherNamespace::cxx_mangling_metadata();
assert(m != nullptr);
assert(m->version == halide_filter_metadata_t::VERSION);
printf("Name is: %s\n", m->name);
assert(strcmp(m->name, "cxx_mangling") == 0);
static_assert(HalideTest::AnotherNamespace::cxx_mangling_argument_info().at(21).name == "output");
int ptr_arg = 42;
int *int_ptr = &ptr_arg;
const int *const_int_ptr = &ptr_arg;
void *void_ptr = nullptr;
const void *const_void_ptr = nullptr;
std::string *string_ptr = nullptr;
const std::string *const_string_ptr = nullptr;
#ifdef TEST_CUDA
// Don't bother calling this (we haven't linked in the CUDA support it needs),
// just force a reference to ensure it is linked in.
int (*f)(halide_buffer_t *,
int8_t, uint8_t,
int16_t, uint16_t,
int32_t, uint32_t,
int64_t, uint64_t,
bool,
float, double,
int32_t *, int32_t const *,
void *, void const *,
void *, void const *,
::my_namespace::my_class const *,
struct ::my_namespace::my_subnamespace::my_struct const *,
my_union const *,
halide_buffer_t *) = HalideTest::cxx_mangling_gpu;
printf("HalideTest::cxx_mangling is at: %p\n", (void *)f);
#else
// TODO: split this up and link CUDA
printf("TEST_CUDA is disabled, skipping cxx_mangling_gpu test.\n");
#endif
my_namespace::my_class mc;
my_namespace::my_subnamespace::my_struct ms;
my_union mu;
int r = HalideTest::AnotherNamespace::cxx_mangling(
input, -1, 0xff, -1, 0xffff, -1, 0xffffffff,
-1, 0xffffffffffffffffLL, true, 42.0, 4239.0f,
int_ptr, const_int_ptr, void_ptr, const_void_ptr,
string_ptr, const_string_ptr,
&mc, &ms, &mu, result);
if (r != 0) {
fprintf(stderr, "Failure!\n");
exit(1);
}
printf("Success!\n");
return 0;
}
|
