ppcg.c

   1 /*
   2  * Copyright 2011      INRIA Saclay
   3  *
   4  * Use of this software is governed by the GNU LGPLv2.1 license
   5  *
   6  * Written by Sven Verdoolaege, INRIA Saclay - Ile-de-France,
   7  * Parc Club Orsay Universite, ZAC des vignes, 4 rue Jacques Monod,
   8  * 91893 Orsay, France
   9  */
  10
  11 #include <assert.h>
  12 #include <stdio.h>
  13 #include <isl/ctx.h>
  14 #include <isl/options.h>
  15 #include <isl/schedule.h>
  16 #include <pet.h>
  17 #include "ppcg_options.h"
  18 #include "cuda.h"
  19 #include "cpu.h"
  20
  21 struct options {
  22         struct isl_options *isl;
  23         struct pet_options *pet;
  24         struct ppcg_options *ppcg;
  25         char *input;
  26 };
  27
  28 ISL_ARGS_START(struct options, options_args)
  29 ISL_ARG_CHILD(struct options, isl, "isl", &isl_options_args, "isl options")
  30 ISL_ARG_CHILD(struct options, pet, "pet", &pet_options_args, "pet options")
  31 ISL_ARG_CHILD(struct options, ppcg, NULL, &ppcg_options_args, "ppcg options")
  32 ISL_ARG_ARG(struct options, input, "input", NULL)
  33 ISL_ARGS_END
  34
  35 ISL_ARG_DEF(options, struct options, options_args)
  36
  37 int main(int argc, char **argv)
  38 {
  39         int r;
  40         isl_ctx *ctx;
  41         struct options *options;
  42         struct pet_scop *scop;
  43
  44         options = options_new_with_defaults();
  45         assert(options);
  46
  47         ctx = isl_ctx_alloc_with_options(&options_args, options);
  48         isl_options_set_schedule_outer_zero_distance(ctx, 1);
  49         argc = options_parse(options, argc, argv, ISL_ARG_ALL);
  50
  51         scop = pet_scop_extract_from_C_source(ctx, options->input, NULL);
  52
  53         if (options->ppcg->target == PPCG_TARGET_CUDA)
  54                 r = cuda_pet(ctx, scop, options->ppcg, options->input);
  55         else
  56                 r = generate_cpu(ctx, scop, options->ppcg, options->input);
  57
  58         pet_scop_free(scop);
  59
  60         isl_ctx_free(ctx);
  61
  62         return r;
  63 }