gcc/testsuite/jit.dg/test-sum-of-squares.c

   1 #include <stdlib.h>
   2 #include <stdio.h>
   3 #include <string.h>
   4
   5 #include "libgccjit.h"
   6
   7 #include "harness.h"
   8
   9 static char *dump_vrp1;
  10
  11 void
  12 create_code (gcc_jit_context *ctxt, void *user_data)
  13 {
  14   /*
  15     Simple sum-of-squares, to test conditionals and looping
  16
  17     int loop_test (int n)
  18     {
  19       int i;
  20       int sum = 0;
  21       for (i = 0; i < n ; i ++)
  22       {
  23         sum += i * i;
  24       }
  25       return sum;
  26    */
  27   gcc_jit_context_enable_dump (ctxt, "tree-vrp1", &dump_vrp1);
  28
  29   gcc_jit_type *the_type =
  30     gcc_jit_context_get_type (ctxt, GCC_JIT_TYPE_INT);
  31   gcc_jit_type *return_type = the_type;
  32
  33   gcc_jit_param *n =
  34     gcc_jit_context_new_param (ctxt, NULL, the_type, "n");
  35   gcc_jit_param *params[1] = {n};
  36   gcc_jit_function *func =
  37     gcc_jit_context_new_function (ctxt, NULL,
  38                                   GCC_JIT_FUNCTION_EXPORTED,
  39                                   return_type,
  40                                   "loop_test",
  41                                   1, params, 0);
  42
  43   /* Build locals:  */
  44   gcc_jit_lvalue *i =
  45     gcc_jit_function_new_local (func, NULL, the_type, "i");
  46   gcc_jit_lvalue *sum =
  47     gcc_jit_function_new_local (func, NULL, the_type, "sum");
  48
  49   gcc_jit_block *initial =
  50     gcc_jit_function_new_block (func, "initial");
  51   gcc_jit_block *loop_cond =
  52     gcc_jit_function_new_block (func, "loop_cond");
  53   gcc_jit_block *loop_body =
  54     gcc_jit_function_new_block (func, "loop_body");
  55   gcc_jit_block *after_loop =
  56     gcc_jit_function_new_block (func, "after_loop");
  57
  58   /* sum = 0; */
  59   gcc_jit_block_add_assignment (
  60     initial, NULL,
  61     sum,
  62     gcc_jit_context_new_rvalue_from_int (ctxt, the_type, 0));
  63
  64   /* i = 0; */
  65   gcc_jit_block_add_assignment (
  66     initial, NULL,
  67     i,
  68     gcc_jit_context_new_rvalue_from_int (ctxt, the_type, 0));
  69
  70   gcc_jit_block_end_with_jump (initial, NULL, loop_cond);
  71
  72   /* if (i >= n) */
  73   gcc_jit_block_end_with_conditional (
  74     loop_cond, NULL,
  75     gcc_jit_context_new_comparison (
  76        ctxt, NULL,
  77        GCC_JIT_COMPARISON_GE,
  78        gcc_jit_lvalue_as_rvalue (i),
  79        gcc_jit_param_as_rvalue (n)),
  80     after_loop,
  81     loop_body);
  82
  83   /* sum += i * i */
  84   gcc_jit_block_add_assignment (
  85     loop_body, NULL,
  86     sum,
  87     gcc_jit_context_new_binary_op (
  88       ctxt, NULL,
  89       GCC_JIT_BINARY_OP_PLUS, the_type,
  90       gcc_jit_lvalue_as_rvalue (sum),
  91       gcc_jit_context_new_binary_op (
  92          ctxt, NULL,
  93          GCC_JIT_BINARY_OP_MULT, the_type,
  94          gcc_jit_lvalue_as_rvalue (i),
  95          gcc_jit_lvalue_as_rvalue (i))));
  96
  97   /* i++ */
  98   gcc_jit_block_add_assignment (
  99     loop_body, NULL,
 100     i,
 101     gcc_jit_context_new_binary_op (
 102       ctxt, NULL,
 103       GCC_JIT_BINARY_OP_PLUS, the_type,
 104       gcc_jit_lvalue_as_rvalue (i),
 105       gcc_jit_context_new_rvalue_from_int (
 106         ctxt,
 107         the_type,
 108         1)));
 109
 110   gcc_jit_block_end_with_jump (loop_body, NULL, loop_cond);
 111
 112   /* return sum */
 113   gcc_jit_block_end_with_return (
 114     after_loop,
 115     NULL,
 116     gcc_jit_lvalue_as_rvalue (sum));
 117 }
 118
 119 void
 120 verify_code (gcc_jit_context *ctxt, gcc_jit_result *result)
 121 {
 122   typedef int (*loop_test_fn_type) (int);
 123   CHECK_NON_NULL (result);
 124   loop_test_fn_type loop_test =
 125     (loop_test_fn_type)gcc_jit_result_get_code (result, "loop_test");
 126   CHECK_NON_NULL (loop_test);
 127   int val = loop_test (10);
 128   note ("loop_test returned: %d", val);
 129   CHECK_VALUE (val, 285);
 130
 131   CHECK_NON_NULL (dump_vrp1);
 132   /* PR jit/64166
 133      An example of using gcc_jit_context_enable_dump to verify a property
 134      of the compile.
 135
 136      In particular, this example verifies that we can read the dump from a
 137      pass with multiple instances.  Given that the format of the VRP dump
 138      is subject to change, merely verify that it contains the function
 139      name.  */
 140   CHECK_STRING_CONTAINS (dump_vrp1, ";; Function loop_test");
 141   free (dump_vrp1);
 142 }