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
|
#include "Halide.h"
#include "halide_test_dirs.h"
#include <cstdio>
using namespace Halide;
void testCompileToOutput(Func j) {
std::string fn_object = Internal::get_test_tmp_dir() + "compile_to_native.o";
printf("fn_object is %s\n", fn_object.c_str());
Internal::ensure_no_file_exists(fn_object);
std::vector<Argument> empty_args;
j.compile_to({{OutputFileType::object, fn_object}}, empty_args, "");
Internal::assert_file_exists(fn_object);
}
void testCompileToOutputAndAssembly(Func j) {
std::string fn_object = Internal::get_test_tmp_dir() + "compile_to_native1.o";
std::string fn_assembly = Internal::get_test_tmp_dir() + "compile_to_assembly1.s";
Internal::ensure_no_file_exists(fn_object);
Internal::ensure_no_file_exists(fn_assembly);
std::vector<Argument> empty_args;
j.compile_to({{OutputFileType::object, fn_object}, {OutputFileType::assembly, fn_assembly}}, empty_args, "");
Internal::assert_file_exists(fn_object);
Internal::assert_file_exists(fn_assembly);
}
int main(int argc, char **argv) {
Func f, g, h, j;
Var x, y;
f(x, y) = x + y;
g(x, y) = cast<float>(f(x, y) + f(x + 1, y));
h(x, y) = f(x, y) + g(x, y);
j(x, y) = h(x, y) * 2;
f.compute_root();
g.compute_root();
h.compute_root();
testCompileToOutput(j);
testCompileToOutputAndAssembly(j);
printf("Success!\n");
return 0;
}
|
