pointer-tasks.c 2.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283
  1. /* GCC-StarPU
  2. Copyright (C) 2011 Institut National de Recherche en Informatique et Automatique
  3. GCC-StarPU is free software: you can redistribute it and/or modify
  4. it under the terms of the GNU General Public License as published by
  5. the Free Software Foundation, either version 3 of the License, or
  6. (at your option) any later version.
  7. GCC-StarPU is distributed in the hope that it will be useful,
  8. but WITHOUT ANY WARRANTY; without even the implied warranty of
  9. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  10. GNU General Public License for more details.
  11. You should have received a copy of the GNU General Public License
  12. along with GCC-StarPU. If not, see <http://www.gnu.org/licenses/>. */
  13. /* (instructions run (ldflags "-lstarpu")) */
  14. #undef NDEBUG
  15. #include <stdlib.h>
  16. #include <assert.h>
  17. #include <string.h>
  18. /* The task under test. */
  19. static void my_pointer_task (const int *x, char a, long long *y, int b)
  20. __attribute__ ((task));
  21. static void my_pointer_task_cpu (const int *x, char a, long long *y, int b)
  22. __attribute__ ((task_implementation ("cpu", my_pointer_task), noinline));
  23. static int implementations_called;
  24. /* The input arguments. */
  25. static const int pointer_arg1[] = { 42, 1, 2, 3, 4, 5 };
  26. static long long *pointer_arg2;
  27. static void
  28. my_pointer_task_cpu (const int *x, char a, long long *y, int b)
  29. {
  30. implementations_called |= STARPU_CPU;
  31. assert (x == pointer_arg1);
  32. assert (y == pointer_arg2);
  33. assert (a == 'S');
  34. assert (b == 42);
  35. }
  36. int
  37. main (int argc, char *argv[])
  38. {
  39. #define COUNT 100
  40. pointer_arg2 = malloc (COUNT * sizeof *pointer_arg2);
  41. memset (pointer_arg2, 0x77, COUNT * sizeof *pointer_arg2);
  42. /* FIXME: Eventually remove calls to libstarpu and use pragmas or generated
  43. code. */
  44. starpu_init (NULL);
  45. /* Register POINTER_ARG1 and POINTER_ARG2. */
  46. #pragma starpu register pointer_arg1
  47. #pragma starpu register pointer_arg2 COUNT
  48. /* Invoke the task, which should make sure it gets called with
  49. EXPECTED. */
  50. my_pointer_task (pointer_arg1, 'S', pointer_arg2, 42);
  51. #pragma starpu wait
  52. assert (implementations_called == STARPU_CPU);
  53. starpu_shutdown ();
  54. free (pointer_arg2);
  55. return EXIT_SUCCESS;
  56. #undef COUNT
  57. }