| #include <stdlib.h> |
| #include <stdio.h> |
| #include <string.h> |
| |
| #include "libgccjit.h" |
| |
| #include "harness.h" |
| |
| static char *dump_vrp1; |
| |
| void |
| create_code (gcc_jit_context *ctxt, void *user_data) |
| { |
| /* |
| Simple sum-of-squares, to test conditionals and looping |
| |
| int loop_test (int n) |
| { |
| int i; |
| int sum = 0; |
| for (i = 0; i < n ; i ++) |
| { |
| sum += i * i; |
| } |
| return sum; |
| */ |
| gcc_jit_context_enable_dump (ctxt, "tree-vrp1", &dump_vrp1); |
| |
| gcc_jit_type *the_type = |
| gcc_jit_context_get_type (ctxt, GCC_JIT_TYPE_INT); |
| gcc_jit_type *return_type = the_type; |
| |
| gcc_jit_param *n = |
| gcc_jit_context_new_param (ctxt, NULL, the_type, "n"); |
| gcc_jit_param *params[1] = {n}; |
| gcc_jit_function *func = |
| gcc_jit_context_new_function (ctxt, NULL, |
| GCC_JIT_FUNCTION_EXPORTED, |
| return_type, |
| "loop_test", |
| 1, params, 0); |
| |
| /* Build locals: */ |
| gcc_jit_lvalue *i = |
| gcc_jit_function_new_local (func, NULL, the_type, "i"); |
| gcc_jit_lvalue *sum = |
| gcc_jit_function_new_local (func, NULL, the_type, "sum"); |
| |
| gcc_jit_block *initial = |
| gcc_jit_function_new_block (func, "initial"); |
| gcc_jit_block *loop_cond = |
| gcc_jit_function_new_block (func, "loop_cond"); |
| gcc_jit_block *loop_body = |
| gcc_jit_function_new_block (func, "loop_body"); |
| gcc_jit_block *after_loop = |
| gcc_jit_function_new_block (func, "after_loop"); |
| |
| /* sum = 0; */ |
| gcc_jit_block_add_assignment ( |
| initial, NULL, |
| sum, |
| gcc_jit_context_new_rvalue_from_int (ctxt, the_type, 0)); |
| |
| /* i = 0; */ |
| gcc_jit_block_add_assignment ( |
| initial, NULL, |
| i, |
| gcc_jit_context_new_rvalue_from_int (ctxt, the_type, 0)); |
| |
| gcc_jit_block_end_with_jump (initial, NULL, loop_cond); |
| |
| /* if (i >= n) */ |
| gcc_jit_block_end_with_conditional ( |
| loop_cond, NULL, |
| gcc_jit_context_new_comparison ( |
| ctxt, NULL, |
| GCC_JIT_COMPARISON_GE, |
| gcc_jit_lvalue_as_rvalue (i), |
| gcc_jit_param_as_rvalue (n)), |
| after_loop, |
| loop_body); |
| |
| /* sum += i * i */ |
| gcc_jit_block_add_assignment ( |
| loop_body, NULL, |
| sum, |
| gcc_jit_context_new_binary_op ( |
| ctxt, NULL, |
| GCC_JIT_BINARY_OP_PLUS, the_type, |
| gcc_jit_lvalue_as_rvalue (sum), |
| gcc_jit_context_new_binary_op ( |
| ctxt, NULL, |
| GCC_JIT_BINARY_OP_MULT, the_type, |
| gcc_jit_lvalue_as_rvalue (i), |
| gcc_jit_lvalue_as_rvalue (i)))); |
| |
| /* i++ */ |
| gcc_jit_block_add_assignment ( |
| loop_body, NULL, |
| i, |
| gcc_jit_context_new_binary_op ( |
| ctxt, NULL, |
| GCC_JIT_BINARY_OP_PLUS, the_type, |
| gcc_jit_lvalue_as_rvalue (i), |
| gcc_jit_context_new_rvalue_from_int ( |
| ctxt, |
| the_type, |
| 1))); |
| |
| gcc_jit_block_end_with_jump (loop_body, NULL, loop_cond); |
| |
| /* return sum */ |
| gcc_jit_block_end_with_return ( |
| after_loop, |
| NULL, |
| gcc_jit_lvalue_as_rvalue (sum)); |
| } |
| |
| void |
| verify_code (gcc_jit_context *ctxt, gcc_jit_result *result) |
| { |
| typedef int (*loop_test_fn_type) (int); |
| CHECK_NON_NULL (result); |
| loop_test_fn_type loop_test = |
| (loop_test_fn_type)gcc_jit_result_get_code (result, "loop_test"); |
| CHECK_NON_NULL (loop_test); |
| int val = loop_test (10); |
| note ("loop_test returned: %d", val); |
| CHECK_VALUE (val, 285); |
| |
| CHECK_NON_NULL (dump_vrp1); |
| /* PR jit/64166 |
| An example of using gcc_jit_context_enable_dump to verify a property |
| of the compile. |
| |
| In particular, this example verifies that we can read the dump from a |
| pass with multiple instances. Given that the format of the VRP dump |
| is subject to change, merely verify that it contains the function |
| name. */ |
| CHECK_STRING_CONTAINS (dump_vrp1, ";; Function loop_test"); |
| free (dump_vrp1); |
| } |