starpu_wait_all_tasks.c 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139
  1. /*
  2. * StarPU
  3. * Copyright (C) INRIA 2008-2009 (see AUTHORS file)
  4. *
  5. * This program is free software; you can redistribute it and/or modify
  6. * it under the terms of the GNU Lesser General Public License as published by
  7. * the Free Software Foundation; either version 2.1 of the License, or (at
  8. * your option) any later version.
  9. *
  10. * This program is distributed in the hope that it will be useful, but
  11. * WITHOUT ANY WARRANTY; without even the implied warranty of
  12. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  13. *
  14. * See the GNU Lesser General Public License in COPYING.LGPL for more details.
  15. */
  16. #include <starpu.h>
  17. #include <sys/time.h>
  18. #include <pthread.h>
  19. #include <stdio.h>
  20. #include <unistd.h>
  21. static unsigned ntasks = 65536;
  22. static void dummy_func(void *descr[], void *arg)
  23. {
  24. }
  25. static starpu_codelet dummy_codelet =
  26. {
  27. .where = STARPU_CPU|STARPU_CUDA|STARPU_OPENCL|STARPU_GORDON,
  28. .cpu_func = dummy_func,
  29. .cuda_func = dummy_func,
  30. .opencl_func = dummy_func,
  31. #ifdef STARPU_USE_GORDON
  32. .gordon_func = 0, /* this will be defined later */
  33. #endif
  34. .model = NULL,
  35. .nbuffers = 0
  36. };
  37. static void init_gordon_kernel(void)
  38. {
  39. #ifdef STARPU_USE_GORDON
  40. unsigned elf_id =
  41. gordon_register_elf_plugin("./microbenchs/null_kernel_gordon.spuelf");
  42. gordon_load_plugin_on_all_spu(elf_id);
  43. unsigned gordon_null_kernel =
  44. gordon_register_kernel(elf_id, "empty_kernel");
  45. gordon_load_kernel_on_all_spu(gordon_null_kernel);
  46. dummy_codelet.gordon_func = gordon_null_kernel;
  47. #endif
  48. }
  49. static void inject_one_task(void)
  50. {
  51. struct starpu_task *task = starpu_task_create();
  52. task->cl = &dummy_codelet;
  53. task->cl_arg = NULL;
  54. task->callback_func = NULL;
  55. task->callback_arg = NULL;
  56. int ret = starpu_task_submit(task);
  57. STARPU_ASSERT(!ret);
  58. }
  59. static struct starpu_conf conf = {
  60. .sched_policy_name = NULL,
  61. .ncpus = -1,
  62. .ncuda = -1,
  63. .nopencl = -1,
  64. .nspus = -1,
  65. .use_explicit_workers_bindid = 0,
  66. .use_explicit_workers_cuda_gpuid = 0,
  67. .use_explicit_workers_opencl_gpuid = 0,
  68. .calibrate = 0
  69. };
  70. static void usage(char **argv)
  71. {
  72. fprintf(stderr, "%s [-i ntasks] [-p sched_policy] [-h]\n", argv[0]);
  73. exit(-1);
  74. }
  75. static void parse_args(int argc, char **argv)
  76. {
  77. int c;
  78. while ((c = getopt(argc, argv, "i:p:h")) != -1)
  79. switch(c) {
  80. case 'i':
  81. ntasks = atoi(optarg);
  82. break;
  83. case 'p':
  84. conf.sched_policy_name = optarg;
  85. break;
  86. case 'h':
  87. usage(argv);
  88. break;
  89. }
  90. }
  91. int main(int argc, char **argv)
  92. {
  93. unsigned i;
  94. double timing;
  95. struct timeval start;
  96. struct timeval end;
  97. parse_args(argc, argv);
  98. starpu_init(&conf);
  99. init_gordon_kernel();
  100. fprintf(stderr, "#tasks : %d\n", ntasks);
  101. gettimeofday(&start, NULL);
  102. for (i = 0; i < ntasks; i++)
  103. {
  104. inject_one_task();
  105. }
  106. starpu_wait_all_tasks();
  107. gettimeofday(&end, NULL);
  108. timing = (double)((end.tv_sec - start.tv_sec)*1000000 + (end.tv_usec - start.tv_usec));
  109. fprintf(stderr, "Total: %lf secs\n", timing/1000000);
  110. fprintf(stderr, "Per task: %lf usecs\n", timing/ntasks);
  111. starpu_shutdown();
  112. return 0;
  113. }