Home Explore Help
Sign In
exa2pro
/
starpu-max
Watch 19
Star 0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Tree: e577005007
Branches Tags
starpu-max
/
mpi
/
include
/
starpu_mpi.h

starpu_mpi.h 4.4 KB
History Raw

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687 
  1. /* StarPU --- Runtime system for heterogeneous multicore architectures.
    2. 

  2.  *
    3. 

  3.  * Copyright (C) 2009-2012, 2014  Université de Bordeaux 1
    4. 

  4.  * Copyright (C) 2010, 2011, 2012, 2013, 2014  Centre National de la Recherche Scientifique
    5. 

  5.  *
    6. 

  6.  * StarPU is free software; you can redistribute it and/or modify
    7. 

  7.  * it under the terms of the GNU Lesser General Public License as published by
    8. 

  8.  * the Free Software Foundation; either version 2.1 of the License, or (at
    9. 

  9.  * your option) any later version.
    10. 

  10.  *
    11. 

  11.  * StarPU is distributed in the hope that it will be useful, but
    12. 

  12.  * WITHOUT ANY WARRANTY; without even the implied warranty of
    13. 

  13.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
    14. 

  14.  *
    15. 

  15.  * See the GNU Lesser General Public License in COPYING.LGPL for more details.
    16. 

  16.  */
    17. 

  17. 

  18. #ifndef __STARPU_MPI_H__
    19. 

  19. #define __STARPU_MPI_H__
    20. 

  20. 

  21. #include <starpu.h>
    22. 

  22. 

  23. #if defined(STARPU_USE_MPI)
    24. 

  24. 

  25. #include <mpi.h>
    26. 

  26. 

  27. #ifdef __cplusplus
    28. 

  28. extern "C" {
    29. 

  29. #endif
    30. 

  30. 

  31. typedef void *starpu_mpi_req;
    32. 

  32. 

  33. int starpu_mpi_isend(starpu_data_handle_t data_handle, starpu_mpi_req *req, int dest, int mpi_tag, MPI_Comm comm);
    34. 

  34. int starpu_mpi_irecv(starpu_data_handle_t data_handle, starpu_mpi_req *req, int source, int mpi_tag, MPI_Comm comm);
    35. 

  35. int starpu_mpi_send(starpu_data_handle_t data_handle, int dest, int mpi_tag, MPI_Comm comm);
    36. 

  36. int starpu_mpi_recv(starpu_data_handle_t data_handle, int source, int mpi_tag, MPI_Comm comm, MPI_Status *status);
    37. 

  37. int starpu_mpi_isend_detached(starpu_data_handle_t data_handle, int dest, int mpi_tag, MPI_Comm comm, void (*callback)(void *), void *arg);
    38. 

  38. int starpu_mpi_irecv_detached(starpu_data_handle_t data_handle, int source, int mpi_tag, MPI_Comm comm, void (*callback)(void *), void *arg);
    39. 

  39. int starpu_mpi_wait(starpu_mpi_req *req, MPI_Status *status);
    40. 

  40. int starpu_mpi_test(starpu_mpi_req *req, int *flag, MPI_Status *status);
    41. 

  41. int starpu_mpi_barrier(MPI_Comm comm);
    42. 

  42. 

  43. int starpu_mpi_irecv_detached_sequential_consistency(starpu_data_handle_t data_handle, int source, int mpi_tag, MPI_Comm comm, void (*callback)(void *), void *arg, int sequential_consistency);
    44. 

  44. 

  45. int starpu_mpi_init(int *argc, char ***argv, int initialize_mpi);
    46. 

  46. int starpu_mpi_initialize(void) STARPU_DEPRECATED;
    47. 

  47. int starpu_mpi_initialize_extended(int *rank, int *world_size) STARPU_DEPRECATED;
    48. 

  48. int starpu_mpi_shutdown(void);
    49. 

  49. 

  50. struct starpu_task *starpu_mpi_task_build(MPI_Comm comm, struct starpu_codelet *codelet, ...);
    51. 

  51. int starpu_mpi_task_post_build(MPI_Comm comm, struct starpu_codelet *codelet, ...);
    52. 

  52. int starpu_mpi_task_insert(MPI_Comm comm, struct starpu_codelet *codelet, ...);
    53. 

  53. /* the function starpu_mpi_insert_task has the same semantics as starpu_mpi_task_insert, it is kept to avoid breaking old codes */
    54. 

  54. int starpu_mpi_insert_task(MPI_Comm comm, struct starpu_codelet *codelet, ...);
    55. 

  55. 

  56. void starpu_mpi_get_data_on_node(MPI_Comm comm, starpu_data_handle_t data_handle, int node);
    57. 

  57. void starpu_mpi_get_data_on_node_detached(MPI_Comm comm, starpu_data_handle_t data_handle, int node, void (*callback)(void*), void *arg);
    58. 

  58. void starpu_mpi_redux_data(MPI_Comm comm, starpu_data_handle_t data_handle);
    59. 

  59. 

  60. int starpu_mpi_scatter_detached(starpu_data_handle_t *data_handles, int count, int root, MPI_Comm comm, void (*scallback)(void *), void *sarg, void (*rcallback)(void *), void *rarg);
    61. 

  61. int starpu_mpi_gather_detached(starpu_data_handle_t *data_handles, int count, int root, MPI_Comm comm, void (*scallback)(void *), void *sarg, void (*rcallback)(void *), void *rarg);
    62. 

  62. 

  63. int starpu_mpi_isend_detached_unlock_tag(starpu_data_handle_t data_handle, int dest, int mpi_tag, MPI_Comm comm, starpu_tag_t tag);
    64. 

  64. int starpu_mpi_irecv_detached_unlock_tag(starpu_data_handle_t data_handle, int source, int mpi_tag, MPI_Comm comm, starpu_tag_t tag);
    65. 

  65. 

  66. int starpu_mpi_isend_array_detached_unlock_tag(unsigned array_size, starpu_data_handle_t *data_handle, int *dest, int *mpi_tag, MPI_Comm *comm, starpu_tag_t tag);
    67. 

  67. int starpu_mpi_irecv_array_detached_unlock_tag(unsigned array_size, starpu_data_handle_t *data_handle, int *source, int *mpi_tag, MPI_Comm *comm, starpu_tag_t tag);
    68. 

  68. 

  69. void starpu_mpi_comm_amounts_retrieve(size_t *comm_amounts);
    70. 

  70. 

  71. void starpu_mpi_cache_flush(MPI_Comm comm, starpu_data_handle_t data_handle);
    72. 

  72. void starpu_mpi_cache_flush_all_data(MPI_Comm comm);
    73. 

  73. 

  74. int starpu_mpi_world_rank(void);
    75. 

  75. 

  76. int starpu_mpi_get_communication_tag(void);
    77. 

  77. void starpu_mpi_set_communication_tag(int tag);
    78. 

  78. 

  79. void starpu_mpi_data_register(starpu_data_handle_t data_handle,int tag, int rank);
    80. 

  80. 

  81. #ifdef __cplusplus
    82. 

  82. }
    83. 

  83. #endif
    84. 

  84. 

  85. #endif // STARPU_USE_MPI
    86. 

  86. #endif // __STARPU_MPI_H__
    87. 

  87.