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
|
#include "Halide.h"
#include "halide_test_dirs.h"
#include <fstream>
#include <iostream>
#include <cassert>
#include <cstdio>
using namespace Halide;
int main(int argc, char **argv) {
if (get_jit_target_from_environment().arch == Target::WebAssembly) {
printf("[SKIP] Skipping test for WebAssembly as it does not support ExternalCode::bitcode_wrapper().\n");
return 0;
}
Var x("x"), y("y");
Func f("f");
f(x, y) = 42;
Target target = get_jit_target_from_environment();
std::string bitcode_file = Internal::get_test_tmp_dir() + "extern.bc";
f.compile_to_bitcode(bitcode_file, {}, "extern", target);
std::vector<uint8_t> bitcode;
std::ifstream bitcode_stream(bitcode_file, std::ios::in | std::ios::binary);
bitcode_stream.seekg(0, std::ios::end);
bitcode.resize(bitcode_stream.tellg());
bitcode_stream.seekg(0, std::ios::beg);
bitcode_stream.read(reinterpret_cast<char *>(&bitcode[0]), bitcode.size());
ExternalCode external_code =
ExternalCode::bitcode_wrapper(target, bitcode, "extern");
Func f_extern;
f_extern.define_extern("extern", {}, type_of<int32_t>(), 2);
Func result;
result(x, y) = f_extern(x, y);
Module module = result.compile_to_module({}, "forty_two", target);
module.append(external_code);
auto forty_two = module.get_function_by_name("forty_two");
Internal::JITModule jit_module(module, forty_two, {});
auto main_function = (int (*)(halide_buffer_t * buf)) jit_module.main_function();
Buffer<int32_t> buf(16, 16);
int ret_code = main_function(buf.raw_buffer());
assert(ret_code == 0);
for (int i = 0; i < 16; i++) {
for (int j = 0; j < 16; j++) {
assert(buf(i, j) == 42);
}
}
printf("Success!\n");
return 0;
}
|
