starpu_mpi_private.c 2.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  1. /* StarPU --- Runtime system for heterogeneous multicore architectures.
  2. *
  3. * Copyright (C) 2013 Inria
  4. * Copyright (C) 2010-2013,2015-2017 CNRS
  5. * Copyright (C) 2010,2012,2014-2016,2018,2020 Université de Bordeaux
  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. #include <starpu_mpi_private.h>
  19. int _starpu_debug_rank=-1;
  20. int _starpu_debug_level_min=0;
  21. int _starpu_debug_level_max=0;
  22. int _starpu_mpi_tag = 42;
  23. int _starpu_mpi_comm_debug;
  24. int _starpu_mpi_thread_cpuid = -1;
  25. int _starpu_mpi_use_prio = 1;
  26. int _starpu_mpi_fake_world_size = -1;
  27. int _starpu_mpi_fake_world_rank = -1;
  28. int _starpu_mpi_use_coop_sends = 1;
  29. int _starpu_mpi_mem_throttle = 0;
  30. void _starpu_mpi_set_debug_level_min(int level)
  31. {
  32. _starpu_debug_level_min = level;
  33. }
  34. void _starpu_mpi_set_debug_level_max(int level)
  35. {
  36. _starpu_debug_level_max = level;
  37. }
  38. int starpu_mpi_get_communication_tag(void)
  39. {
  40. return _starpu_mpi_tag;
  41. }
  42. void starpu_mpi_set_communication_tag(int tag)
  43. {
  44. _starpu_mpi_tag = tag;
  45. }
  46. char *_starpu_mpi_get_mpi_error_code(int code)
  47. {
  48. static char str[MPI_MAX_OBJECT_NAME];
  49. int len;
  50. MPI_Error_string(code, str, &len);
  51. return str;
  52. }
  53. void _starpu_mpi_env_init(void)
  54. {
  55. _starpu_mpi_comm_debug = starpu_getenv("STARPU_MPI_COMM") != NULL;
  56. _starpu_mpi_fake_world_size = starpu_get_env_number("STARPU_MPI_FAKE_SIZE");
  57. _starpu_mpi_fake_world_rank = starpu_get_env_number("STARPU_MPI_FAKE_RANK");
  58. _starpu_mpi_thread_cpuid = starpu_get_env_number_default("STARPU_MPI_THREAD_CPUID", -1);
  59. _starpu_mpi_use_prio = starpu_get_env_number_default("STARPU_MPI_PRIORITIES", 1);
  60. _starpu_mpi_use_coop_sends = starpu_get_env_number_default("STARPU_MPI_COOP_SENDS", 1);
  61. _starpu_mpi_mem_throttle = starpu_get_env_number_default("STARPU_MPI_MEM_THROTTLE", 0);
  62. }