static_restartable_using_initializer.c 2.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394
  1. /* StarPU --- Runtime system for heterogeneous multicore architectures.
  2. *
  3. * Copyright (C) 2009, 2010-2011 Université de Bordeaux 1
  4. * Copyright (C) 2010, 2011 Centre National de la Recherche Scientifique
  5. *
  6. * StarPU is free software; you can redistribute it and/or modify
  7. * it under the terms of the GNU Lesser General Public License as published by
  8. * the Free Software Foundation; either version 2.1 of the License, or (at
  9. * your option) any later version.
  10. *
  11. * StarPU is distributed in the hope that it will be useful, but
  12. * WITHOUT ANY WARRANTY; without even the implied warranty of
  13. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  14. *
  15. * See the GNU Lesser General Public License in COPYING.LGPL for more details.
  16. */
  17. #include <sys/time.h>
  18. #include <stdio.h>
  19. #include <unistd.h>
  20. #include <starpu.h>
  21. #define FPRINTF(ofile, fmt, args ...) do { if (!getenv("STARPU_SSILENT")) {fprintf(ofile, fmt, ##args); }} while(0)
  22. /* This is equivalent to calling starpu_task_init later on */
  23. struct starpu_task task = STARPU_TASK_INITIALIZER;
  24. static unsigned ntasks = 65536;
  25. static void dummy_func(void *descr[] __attribute__ ((unused)), void *arg __attribute__ ((unused)))
  26. {
  27. }
  28. static starpu_codelet dummy_codelet =
  29. {
  30. .where = STARPU_CPU|STARPU_CUDA|STARPU_OPENCL,
  31. .cpu_func = dummy_func,
  32. .cuda_func = dummy_func,
  33. .opencl_func = dummy_func,
  34. .model = NULL,
  35. .nbuffers = 0
  36. };
  37. static void parse_args(int argc, char **argv)
  38. {
  39. int c;
  40. while ((c = getopt(argc, argv, "i:")) != -1)
  41. switch(c) {
  42. case 'i':
  43. ntasks = atoi(optarg);
  44. break;
  45. }
  46. }
  47. int main(int argc, char **argv)
  48. {
  49. unsigned i;
  50. double timing;
  51. struct timeval start;
  52. struct timeval end;
  53. parse_args(argc, argv);
  54. starpu_init(NULL);
  55. task.cl = &dummy_codelet;
  56. task.detach = 0;
  57. FPRINTF(stderr, "#tasks : %u\n", ntasks);
  58. gettimeofday(&start, NULL);
  59. for (i = 0; i < ntasks; i++)
  60. {
  61. starpu_task_submit(&task);
  62. starpu_task_wait(&task);
  63. }
  64. gettimeofday(&end, NULL);
  65. timing = (double)((end.tv_sec - start.tv_sec)*1000000
  66. + (end.tv_usec - start.tv_usec));
  67. FPRINTF(stderr, "Total: %f secs\n", timing/1000000);
  68. FPRINTF(stderr, "Per task: %f usecs\n", timing/ntasks);
  69. starpu_shutdown();
  70. return 0;
  71. }