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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include "libgccjit.h"
#include "harness.h"
void
create_code (gcc_jit_context *ctxt, void *user_data)
{
/* Let's try to inject the equivalent of:
double
my_dot_product (int n, double *a, double *b)
{
double result = 0.;
for (int i = 0; i < n; i++)
result += a[i] * b[i];
return result
}
and see what the optimizer can do. */
gcc_jit_type *val_type =
gcc_jit_context_get_type (ctxt, GCC_JIT_TYPE_DOUBLE);
gcc_jit_type *ptr_type = gcc_jit_type_get_pointer (val_type);
gcc_jit_type *int_type =
gcc_jit_context_get_type (ctxt, GCC_JIT_TYPE_INT);
gcc_jit_type *return_type = val_type;
gcc_jit_param *param_n =
gcc_jit_context_new_param (ctxt, NULL, int_type, "n");
gcc_jit_param *param_a =
gcc_jit_context_new_param (ctxt, NULL, ptr_type, "a");
gcc_jit_param *param_b =
gcc_jit_context_new_param (ctxt, NULL, ptr_type, "b");
gcc_jit_param *params[3] = {param_n, param_a, param_b};
gcc_jit_function *func =
gcc_jit_context_new_function (ctxt, NULL,
GCC_JIT_FUNCTION_EXPORTED,
return_type,
"my_dot_product",
3, params, 0);
gcc_jit_block *initial = gcc_jit_function_new_block (func, "initial");
gcc_jit_block *loop_test = gcc_jit_function_new_block (func, "loop_test");
gcc_jit_block *loop_body = gcc_jit_function_new_block (func, "loop_body");
gcc_jit_block *final = gcc_jit_function_new_block (func, "final");
/* Build: "double result = 0.;" */
gcc_jit_lvalue *result =
gcc_jit_function_new_local (func, NULL, val_type, "result");
gcc_jit_block_add_assignment (initial, NULL,
result, gcc_jit_context_zero (ctxt, val_type));
/* Build: "for (int i = 0; i < n; i++)" */
gcc_jit_lvalue *i =
gcc_jit_function_new_local (func, NULL, int_type, "i");
gcc_jit_block_add_assignment (initial, NULL,
i, gcc_jit_context_zero (ctxt, int_type));
gcc_jit_block_end_with_jump (initial, NULL, loop_test);
gcc_jit_block_end_with_conditional (
loop_test, NULL,
/* (i < n) */
gcc_jit_context_new_comparison (
ctxt, NULL,
GCC_JIT_COMPARISON_LT,
gcc_jit_lvalue_as_rvalue (i),
gcc_jit_param_as_rvalue (param_n)),
loop_body,
final);
/* Build: "result += a[i] * b[i];" */
gcc_jit_block_add_assignment_op (
loop_body, NULL,
result,
GCC_JIT_BINARY_OP_PLUS,
gcc_jit_context_new_binary_op (
ctxt, NULL,
GCC_JIT_BINARY_OP_MULT,
val_type,
gcc_jit_lvalue_as_rvalue (
gcc_jit_context_new_array_access (
ctxt, NULL,
gcc_jit_param_as_rvalue (param_a),
gcc_jit_lvalue_as_rvalue (i))),
gcc_jit_lvalue_as_rvalue (
gcc_jit_context_new_array_access (
ctxt, NULL,
gcc_jit_param_as_rvalue (param_b),
gcc_jit_lvalue_as_rvalue (i)))));
/* Build: "i++" */
gcc_jit_block_add_assignment_op (
loop_body, NULL,
i,
GCC_JIT_BINARY_OP_PLUS,
gcc_jit_context_one (ctxt, int_type));
gcc_jit_block_end_with_jump (loop_body, NULL, loop_test);
/* Build: "return result;" */
gcc_jit_block_end_with_return (
final,
NULL,
gcc_jit_lvalue_as_rvalue (result));
}
void
verify_code (gcc_jit_context *ctxt, gcc_jit_result *result)
{
typedef double (*my_dot_product_fn_type) (int n, double *a, double *b);
CHECK_NON_NULL (result);
my_dot_product_fn_type my_dot_product =
(my_dot_product_fn_type)gcc_jit_result_get_code (result,
"my_dot_product");
CHECK_NON_NULL (my_dot_product);
double test_array[] = {1., 2., 3., 4., 5., 6., 7., 8., 9., 10.};
double val = my_dot_product (10, test_array, test_array);
note ("my_dot_product returned: %f", val);
CHECK_VALUE (val, 385.0);
}
|