starpu_mpi_fxt.h 6.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133
  1. /* StarPU --- Runtime system for heterogeneous multicore architectures.
  2. *
  3. * Copyright (C) 2010, 2017 Université de Bordeaux
  4. * Copyright (C) 2010, 2012, 2016 CNRS
  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. #ifndef __STARPU_MPI_FXT_H__
  18. #define __STARPU_MPI_FXT_H__
  19. #include <starpu.h>
  20. #include <common/config.h>
  21. #include <common/fxt.h>
  22. #ifdef __cplusplus
  23. extern "C" {
  24. #endif
  25. #define _STARPU_MPI_FUT_START 0x5201
  26. #define _STARPU_MPI_FUT_STOP 0x5202
  27. #define _STARPU_MPI_FUT_BARRIER 0x5203
  28. #define _STARPU_MPI_FUT_ISEND_SUBMIT_BEGIN 0x5204
  29. #define _STARPU_MPI_FUT_ISEND_SUBMIT_END 0x5205
  30. #define _STARPU_MPI_FUT_IRECV_SUBMIT_BEGIN 0x5206
  31. #define _STARPU_MPI_FUT_IRECV_SUBMIT_END 0x5207
  32. #define _STARPU_MPI_FUT_ISEND_COMPLETE_BEGIN 0x5208
  33. #define _STARPU_MPI_FUT_ISEND_COMPLETE_END 0x5209
  34. #define _STARPU_MPI_FUT_IRECV_COMPLETE_BEGIN 0x5210
  35. #define _STARPU_MPI_FUT_IRECV_COMPLETE_END 0x5211
  36. #define _STARPU_MPI_FUT_SLEEP_BEGIN 0x5212
  37. #define _STARPU_MPI_FUT_SLEEP_END 0x5213
  38. #define _STARPU_MPI_FUT_DTESTING_BEGIN 0x5214
  39. #define _STARPU_MPI_FUT_DTESTING_END 0x5215
  40. #define _STARPU_MPI_FUT_UTESTING_BEGIN 0x5216
  41. #define _STARPU_MPI_FUT_UTESTING_END 0x5217
  42. #define _STARPU_MPI_FUT_UWAIT_BEGIN 0x5218
  43. #define _STARPU_MPI_FUT_UWAIT_END 0x5219
  44. #define _STARPU_MPI_FUT_DATA_SET_RANK 0x521a
  45. #define _STARPU_MPI_FUT_IRECV_TERMINATED 0x521b
  46. #define _STARPU_MPI_FUT_ISEND_TERMINATED 0x521c
  47. #ifdef STARPU_USE_FXT
  48. #define _STARPU_MPI_TRACE_START(rank, worldsize) \
  49. FUT_DO_PROBE3(_STARPU_MPI_FUT_START, (rank), (worldsize), _starpu_gettid());
  50. #define _STARPU_MPI_TRACE_STOP(rank, worldsize) \
  51. FUT_DO_PROBE3(_STARPU_MPI_FUT_STOP, (rank), (worldsize), _starpu_gettid());
  52. #define _STARPU_MPI_TRACE_BARRIER(rank, worldsize, key) \
  53. FUT_DO_PROBE4(_STARPU_MPI_FUT_BARRIER, (rank), (worldsize), (key), _starpu_gettid());
  54. #define _STARPU_MPI_TRACE_ISEND_SUBMIT_BEGIN(dest, mpi_tag, size) \
  55. FUT_DO_PROBE4(_STARPU_MPI_FUT_ISEND_SUBMIT_BEGIN, (dest), (mpi_tag), (size), _starpu_gettid());
  56. #define _STARPU_MPI_TRACE_ISEND_SUBMIT_END(dest, mpi_tag, size, jobid) \
  57. FUT_DO_PROBE5(_STARPU_MPI_FUT_ISEND_SUBMIT_END, (dest), (mpi_tag), (size), (jobid), _starpu_gettid());
  58. #define _STARPU_MPI_TRACE_IRECV_SUBMIT_BEGIN(src, mpi_tag) \
  59. FUT_DO_PROBE3(_STARPU_MPI_FUT_IRECV_SUBMIT_BEGIN, (src), (mpi_tag), _starpu_gettid());
  60. #define _STARPU_MPI_TRACE_IRECV_SUBMIT_END(src, mpi_tag) \
  61. FUT_DO_PROBE3(_STARPU_MPI_FUT_IRECV_SUBMIT_END, (src), (mpi_tag), _starpu_gettid());
  62. #define _STARPU_MPI_TRACE_ISEND_COMPLETE_BEGIN(dest, mpi_tag, size) \
  63. FUT_DO_PROBE4(_STARPU_MPI_FUT_ISEND_COMPLETE_BEGIN, (dest), (mpi_tag), (size), _starpu_gettid());
  64. #define _STARPU_MPI_TRACE_COMPLETE_BEGIN(type, rank, mpi_tag) \
  65. if (type == RECV_REQ) { _STARPU_MPI_TRACE_IRECV_COMPLETE_BEGIN((rank), (mpi_tag)); } else if (type == SEND_REQ) { _STARPU_MPI_TRACE_ISEND_COMPLETE_BEGIN((rank), (mpi_tag), 0); }
  66. #define _STARPU_MPI_TRACE_ISEND_COMPLETE_END(dest, mpi_tag, size) \
  67. FUT_DO_PROBE4(_STARPU_MPI_FUT_ISEND_COMPLETE_END, (dest), (mpi_tag), (size), _starpu_gettid());
  68. #define _STARPU_MPI_TRACE_IRECV_COMPLETE_BEGIN(src, mpi_tag) \
  69. FUT_DO_PROBE3(_STARPU_MPI_FUT_IRECV_COMPLETE_BEGIN, (src), (mpi_tag), _starpu_gettid());
  70. #define _STARPU_MPI_TRACE_IRECV_COMPLETE_END(src, mpi_tag) \
  71. FUT_DO_PROBE3(_STARPU_MPI_FUT_IRECV_COMPLETE_END, (src), (mpi_tag), _starpu_gettid());
  72. #define _STARPU_MPI_TRACE_COMPLETE_END(type, rank, mpi_tag) \
  73. if (type == RECV_REQ) { _STARPU_MPI_TRACE_IRECV_COMPLETE_END((rank), (mpi_tag)); } else if (type == SEND_REQ) { _STARPU_MPI_TRACE_ISEND_COMPLETE_END((rank), (mpi_tag), 0); }
  74. #define _STARPU_MPI_TRACE_TERMINATED(req, rank, mpi_tag) \
  75. if ((req)->request_type == RECV_REQ) FUT_DO_PROBE4(_STARPU_MPI_FUT_IRECV_TERMINATED, (rank), (mpi_tag), (req)->post_sync_jobid, _starpu_gettid()); else \
  76. if ((req)->request_type == SEND_REQ) FUT_DO_PROBE3(_STARPU_MPI_FUT_ISEND_TERMINATED, (rank), (mpi_tag), _starpu_gettid());
  77. #define _STARPU_MPI_TRACE_SLEEP_BEGIN() \
  78. FUT_DO_PROBE1(_STARPU_MPI_FUT_SLEEP_BEGIN, _starpu_gettid());
  79. #define _STARPU_MPI_TRACE_SLEEP_END() \
  80. FUT_DO_PROBE1(_STARPU_MPI_FUT_SLEEP_END, _starpu_gettid());
  81. #define _STARPU_MPI_TRACE_DTESTING_BEGIN() \
  82. FUT_DO_PROBE1(_STARPU_MPI_FUT_DTESTING_BEGIN, _starpu_gettid());
  83. #define _STARPU_MPI_TRACE_DTESTING_END() \
  84. FUT_DO_PROBE1(_STARPU_MPI_FUT_DTESTING_END, _starpu_gettid());
  85. #define _STARPU_MPI_TRACE_UTESTING_BEGIN(src, mpi_tag) \
  86. FUT_DO_PROBE3(_STARPU_MPI_FUT_UTESTING_BEGIN, (src), (mpi_tag), _starpu_gettid());
  87. #define _STARPU_MPI_TRACE_UTESTING_END(src, mpi_tag) \
  88. FUT_DO_PROBE3(_STARPU_MPI_FUT_UTESTING_END, (src), (mpi_tag), _starpu_gettid());
  89. #define _STARPU_MPI_TRACE_UWAIT_BEGIN(src, mpi_tag) \
  90. FUT_DO_PROBE3(_STARPU_MPI_FUT_UWAIT_BEGIN, (src), (mpi_tag), _starpu_gettid());
  91. #define _STARPU_MPI_TRACE_UWAIT_END(src, mpi_tag) \
  92. FUT_DO_PROBE3(_STARPU_MPI_FUT_UWAIT_END, (src), (mpi_tag), _starpu_gettid());
  93. #define _STARPU_MPI_TRACE_DATA_SET_RANK(handle, rank) \
  94. FUT_DO_PROBE3(_STARPU_MPI_FUT_DATA_SET_RANK, (handle), (rank), _starpu_gettid());
  95. #define TRACE
  96. #else
  97. #define _STARPU_MPI_TRACE_START(a, b) do {} while(0);
  98. #define _STARPU_MPI_TRACE_STOP(a, b) do {} while(0);
  99. #define _STARPU_MPI_TRACE_BARRIER(a, b, c) do {} while(0);
  100. #define _STARPU_MPI_TRACE_ISEND_SUBMIT_BEGIN(a, b, c) do {} while(0);
  101. #define _STARPU_MPI_TRACE_ISEND_SUBMIT_END(a, b, c, d) do {} while(0);
  102. #define _STARPU_MPI_TRACE_IRECV_SUBMIT_BEGIN(a, b) do {} while(0);
  103. #define _STARPU_MPI_TRACE_IRECV_SUBMIT_END(a, b) do {} while(0);
  104. #define _STARPU_MPI_TRACE_ISEND_COMPLETE_BEGIN(a, b, c) do {} while(0);
  105. #define _STARPU_MPI_TRACE_COMPLETE_BEGIN(a, b, c) do {} while(0);
  106. #define _STARPU_MPI_TRACE_COMPLETE_END(a, b, c) do {} while(0);
  107. #define _STARPU_MPI_TRACE_TERMINATED(a, b, c) do {} while(0);
  108. #define _STARPU_MPI_TRACE_ISEND_COMPLETE_END(a, b, c) do {} while(0);
  109. #define _STARPU_MPI_TRACE_IRECV_COMPLETE_BEGIN(a, b) do {} while(0);
  110. #define _STARPU_MPI_TRACE_IRECV_COMPLETE_END(a, b) do {} while(0);
  111. #define _STARPU_MPI_TRACE_SLEEP_BEGIN() do {} while(0);
  112. #define _STARPU_MPI_TRACE_SLEEP_END() do {} while(0);
  113. #define _STARPU_MPI_TRACE_DTESTING_BEGIN() do {} while(0);
  114. #define _STARPU_MPI_TRACE_DTESTING_END() do {} while(0);
  115. #define _STARPU_MPI_TRACE_UTESTING_BEGIN(a, b) do {} while(0);
  116. #define _STARPU_MPI_TRACE_UTESTING_END(a, b) do {} while(0);
  117. #define _STARPU_MPI_TRACE_UWAIT_BEGIN(a, b) do {} while(0);
  118. #define _STARPU_MPI_TRACE_UWAIT_END(a, b) do {} while(0);
  119. #define _STARPU_MPI_TRACE_DATA_SET_RANK(a, b) do {} while(0);
  120. #endif
  121. #ifdef __cplusplus
  122. }
  123. #endif
  124. #endif // __STARPU_MPI_FXT_H__