starpu_task_util.h 3.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879
  1. /* StarPU --- Runtime system for heterogeneous multicore architectures.
  2. *
  3. * Copyright (C) 2010-2015 Université de Bordeaux
  4. * Copyright (C) 2010-2014, 2016 CNRS
  5. * Copyright (C) 2014 INRIA
  6. *
  7. * StarPU is free software; you can redistribute it and/or modify
  8. * it under the terms of the GNU Lesser General Public License as published by
  9. * the Free Software Foundation; either version 2.1 of the License, or (at
  10. * your option) any later version.
  11. *
  12. * StarPU is distributed in the hope that it will be useful, but
  13. * WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  15. *
  16. * See the GNU Lesser General Public License in COPYING.LGPL for more details.
  17. */
  18. #ifndef __STARPU_TASK_UTIL_H__
  19. #define __STARPU_TASK_UTIL_H__
  20. #include <stdio.h>
  21. #include <stdlib.h>
  22. #include <string.h>
  23. #include <assert.h>
  24. #include <starpu.h>
  25. #ifdef __cplusplus
  26. extern "C"
  27. {
  28. #endif
  29. void starpu_create_sync_task(starpu_tag_t sync_tag, unsigned ndeps, starpu_tag_t *deps, void (*callback)(void *), void *callback_arg);
  30. /* NOTE: when adding a value here, please make sure to update both
  31. * src/util/starpu_task_insert_utils.c (in two places) and
  32. * mpi/src/starpu_mpi_task_insert.c */
  33. #define STARPU_MODE_SHIFT 17
  34. #define STARPU_VALUE (1<<STARPU_MODE_SHIFT)
  35. #define STARPU_CALLBACK (2<<STARPU_MODE_SHIFT)
  36. #define STARPU_CALLBACK_WITH_ARG (3<<STARPU_MODE_SHIFT)
  37. #define STARPU_CALLBACK_ARG (4<<STARPU_MODE_SHIFT)
  38. #define STARPU_PRIORITY (5<<STARPU_MODE_SHIFT)
  39. #define STARPU_EXECUTE_ON_NODE (6<<STARPU_MODE_SHIFT)
  40. #define STARPU_EXECUTE_ON_DATA (7<<STARPU_MODE_SHIFT)
  41. #define STARPU_DATA_ARRAY (8<<STARPU_MODE_SHIFT)
  42. #define STARPU_DATA_MODE_ARRAY (9<<STARPU_MODE_SHIFT)
  43. #define STARPU_TAG (10<<STARPU_MODE_SHIFT)
  44. #define STARPU_HYPERVISOR_TAG (11<<STARPU_MODE_SHIFT)
  45. #define STARPU_FLOPS (12<<STARPU_MODE_SHIFT)
  46. #define STARPU_SCHED_CTX (13<<STARPU_MODE_SHIFT)
  47. #define STARPU_PROLOGUE_CALLBACK (14<<STARPU_MODE_SHIFT)
  48. #define STARPU_PROLOGUE_CALLBACK_ARG (15<<STARPU_MODE_SHIFT)
  49. #define STARPU_PROLOGUE_CALLBACK_POP (16<<STARPU_MODE_SHIFT)
  50. #define STARPU_PROLOGUE_CALLBACK_POP_ARG (17<<STARPU_MODE_SHIFT)
  51. #define STARPU_EXECUTE_ON_WORKER (18<<STARPU_MODE_SHIFT)
  52. #define STARPU_TAG_ONLY (19<<STARPU_MODE_SHIFT)
  53. #define STARPU_POSSIBLY_PARALLEL (20<<STARPU_MODE_SHIFT)
  54. #define STARPU_WORKER_ORDER (21<<STARPU_MODE_SHIFT)
  55. #define STARPU_NODE_SELECTION_POLICY (22<<STARPU_MODE_SHIFT)
  56. #define STARPU_NAME (23<<STARPU_MODE_SHIFT)
  57. #define STARPU_CL_ARGS (24<<STARPU_MODE_SHIFT)
  58. #define STARPU_SHIFTED_MODE_MAX (25<<STARPU_MODE_SHIFT)
  59. struct starpu_task *starpu_task_build(struct starpu_codelet *cl, ...);
  60. int starpu_task_insert(struct starpu_codelet *cl, ...);
  61. /* the function starpu_insert_task has the same semantics as starpu_task_insert, it is kept to avoid breaking old codes */
  62. int starpu_insert_task(struct starpu_codelet *cl, ...);
  63. void starpu_codelet_unpack_args(void *cl_arg, ...);
  64. void starpu_codelet_unpack_args_and_copyleft(void *cl_arg, void *buffer, size_t buffer_size, ...);
  65. void starpu_codelet_pack_args(void **arg_buffer, size_t *arg_buffer_size, ...);
  66. #ifdef __cplusplus
  67. }
  68. #endif
  69. #endif /* __STARPU_TASK_UTIL_H__ */