starpu_mpi.h 1.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  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. #ifndef __STARPU_MPI_H__
  17. #define __STARPU_MPI_H__
  18. #include <starpu.h>
  19. #include <mpi.h>
  20. #include <common/list.h>
  21. #include <pthread.h>
  22. LIST_TYPE(starpu_mpi_req,
  23. void *ptr;
  24. MPI_Datatype datatype;
  25. MPI_Request request;
  26. );
  27. int starpu_mpi_isend(starpu_data_handle data_handle, starpu_mpi_req_t *req,
  28. int dest, int mpi_tag, MPI_Comm comm,
  29. void (*callback)(void *));
  30. int starpu_mpi_irecv(starpu_data_handle data_handle, starpu_mpi_req_t *req,
  31. int source, int mpi_tag, MPI_Comm comm,
  32. void (*callback)(void *));
  33. int starpu_mpi_send(starpu_data_handle data_handle,
  34. int dest, int mpi_tag, MPI_Comm comm);
  35. int starpu_mpi_recv(starpu_data_handle data_handle,
  36. int source, int mpi_tag, MPI_Comm comm, MPI_Status *status);
  37. int starpu_mpi_wait(starpu_mpi_req_t *req);
  38. int starpu_mpi_test(starpu_mpi_req_t *req, int *flag);
  39. int starpu_mpi_initialize(void);
  40. int starpu_mpi_shutdown(void);
  41. #endif // __STARPU_MPI_H__