data_interface.h 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687
  1. /* StarPU --- Runtime system for heterogeneous multicore architectures.
  2. *
  3. * Copyright (C) 2009-2012, 2014-2016 Université de Bordeaux
  4. * Copyright (C) 2010, 2012, 2013, 2014, 2015 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 __DATA_INTERFACE_H__
  19. #define __DATA_INTERFACE_H__
  20. #include <starpu.h>
  21. #include <common/config.h>
  22. #include <common/uthash.h>
  23. #ifdef STARPU_OPENMP
  24. #include <util/openmp_runtime_support.h>
  25. #endif
  26. /* Generic type representing an interface, for now it's only used before
  27. * execution on message-passing devices but it can be useful in other cases.
  28. */
  29. union _starpu_interface
  30. {
  31. struct starpu_matrix_interface matrix;
  32. struct starpu_block_interface block;
  33. struct starpu_vector_interface vector;
  34. struct starpu_csr_interface csr;
  35. struct starpu_coo_interface coo;
  36. struct starpu_bcsr_interface bcsr;
  37. struct starpu_variable_interface variable;
  38. struct starpu_multiformat_interface multiformat;
  39. };
  40. /* Some data interfaces or filters use this interface internally */
  41. extern struct starpu_data_interface_ops starpu_interface_matrix_ops;
  42. extern struct starpu_data_interface_ops starpu_interface_block_ops;
  43. extern struct starpu_data_interface_ops starpu_interface_vector_ops;
  44. extern struct starpu_data_interface_ops starpu_interface_csr_ops;
  45. extern struct starpu_data_interface_ops starpu_interface_bcsr_ops;
  46. extern struct starpu_data_interface_ops starpu_interface_variable_ops;
  47. extern struct starpu_data_interface_ops starpu_interface_void_ops;
  48. extern struct starpu_data_interface_ops starpu_interface_multiformat_ops;
  49. void _starpu_data_free_interfaces(starpu_data_handle_t handle)
  50. STARPU_ATTRIBUTE_INTERNAL;
  51. extern
  52. int _starpu_data_handle_init(starpu_data_handle_t handle, struct starpu_data_interface_ops *interface_ops, unsigned int mf_node);
  53. void _starpu_data_initialize_per_worker(starpu_data_handle_t handle);
  54. extern struct starpu_arbiter *_starpu_global_arbiter;
  55. extern void _starpu_data_interface_init(void) STARPU_ATTRIBUTE_INTERNAL;
  56. extern int __starpu_data_check_not_busy(starpu_data_handle_t handle) STARPU_ATTRIBUTE_INTERNAL STARPU_ATTRIBUTE_WARN_UNUSED_RESULT;
  57. #define _starpu_data_check_not_busy(handle) \
  58. (STARPU_UNLIKELY(!handle->busy_count && \
  59. (handle->busy_waiting || handle->lazy_unregister)) ? \
  60. __starpu_data_check_not_busy(handle) : 0)
  61. extern void _starpu_data_interface_shutdown(void) STARPU_ATTRIBUTE_INTERNAL;
  62. #ifdef STARPU_OPENMP
  63. void _starpu_omp_unregister_region_handles(struct starpu_omp_region *region);
  64. void _starpu_omp_unregister_task_handles(struct starpu_omp_task *task);
  65. #endif
  66. struct starpu_data_interface_ops *_starpu_data_interface_get_ops(unsigned interface_id);
  67. extern void _starpu_data_register_ram_pointer(starpu_data_handle_t handle,
  68. void *ptr)
  69. STARPU_ATTRIBUTE_INTERNAL;
  70. extern void _starpu_data_unregister_ram_pointer(starpu_data_handle_t handle)
  71. STARPU_ATTRIBUTE_INTERNAL;
  72. #define _starpu_data_is_multiformat_handle(handle) handle->ops->is_multiformat
  73. #endif // __DATA_INTERFACE_H__