execution 5.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145
  1. // -*- C++ -*-
  2. //===----------------------------------------------------------------------===//
  3. //
  4. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
  5. // See https://llvm.org/LICENSE.txt for license information.
  6. // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
  7. //
  8. //===----------------------------------------------------------------------===//
  9. #ifndef _LIBCPP_EXECUTION
  10. #define _LIBCPP_EXECUTION
  11. /*
  12. namespace std::execution {
  13. struct sequenced_policy;
  14. struct parallel_policy;
  15. struct parallel_unsequenced_policy;
  16. struct unsequenced_policy; // since C++20
  17. inline constexpr sequenced_policy seq = implementation-defined;
  18. inline constexpr parallel_policy par = implementation-defined;
  19. inline constexpr parallel_unsequenced_policy par_unseq = implementation-defined;
  20. inline constexpr unsequenced_policy unseq = implementation-defined; // since C++20
  21. }
  22. namespace std {
  23. template <class T>
  24. struct is_execution_policy;
  25. template <class T>
  26. inline constexpr bool is_execution_policy_v;
  27. }
  28. */
  29. #include <__config>
  30. #include <__type_traits/is_execution_policy.h>
  31. #include <__type_traits/is_same.h>
  32. #include <__type_traits/remove_cvref.h>
  33. #include <version>
  34. #if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER)
  35. # pragma GCC system_header
  36. #endif
  37. #if !defined(_LIBCPP_HAS_NO_INCOMPLETE_PSTL) && _LIBCPP_STD_VER >= 17
  38. _LIBCPP_BEGIN_NAMESPACE_STD
  39. namespace execution {
  40. struct sequenced_policy {
  41. _LIBCPP_HIDE_FROM_ABI constexpr explicit sequenced_policy(__disable_user_instantiations_tag) {}
  42. sequenced_policy(const sequenced_policy&) = delete;
  43. sequenced_policy& operator=(const sequenced_policy&) = delete;
  44. };
  45. inline constexpr sequenced_policy seq{__disable_user_instantiations_tag{}};
  46. struct parallel_policy {
  47. _LIBCPP_HIDE_FROM_ABI constexpr explicit parallel_policy(__disable_user_instantiations_tag) {}
  48. parallel_policy(const parallel_policy&) = delete;
  49. parallel_policy& operator=(const parallel_policy&) = delete;
  50. };
  51. inline constexpr parallel_policy par{__disable_user_instantiations_tag{}};
  52. struct parallel_unsequenced_policy {
  53. _LIBCPP_HIDE_FROM_ABI constexpr explicit parallel_unsequenced_policy(__disable_user_instantiations_tag) {}
  54. parallel_unsequenced_policy(const parallel_unsequenced_policy&) = delete;
  55. parallel_unsequenced_policy& operator=(const parallel_unsequenced_policy&) = delete;
  56. };
  57. inline constexpr parallel_unsequenced_policy par_unseq{__disable_user_instantiations_tag{}};
  58. struct __unsequenced_policy {
  59. _LIBCPP_HIDE_FROM_ABI constexpr explicit __unsequenced_policy(__disable_user_instantiations_tag) {}
  60. __unsequenced_policy(const __unsequenced_policy&) = delete;
  61. __unsequenced_policy& operator=(const __unsequenced_policy&) = delete;
  62. };
  63. constexpr __unsequenced_policy __unseq{__disable_user_instantiations_tag{}};
  64. # if _LIBCPP_STD_VER >= 20
  65. struct unsequenced_policy {
  66. _LIBCPP_HIDE_FROM_ABI constexpr explicit unsequenced_policy(__disable_user_instantiations_tag) {}
  67. unsequenced_policy(const unsequenced_policy&) = delete;
  68. unsequenced_policy& operator=(const unsequenced_policy&) = delete;
  69. };
  70. inline constexpr unsequenced_policy unseq{__disable_user_instantiations_tag{}};
  71. # endif // _LIBCPP_STD_VER >= 20
  72. } // namespace execution
  73. template <>
  74. inline constexpr bool is_execution_policy_v<execution::sequenced_policy> = true;
  75. template <>
  76. inline constexpr bool is_execution_policy_v<execution::parallel_policy> = true;
  77. template <>
  78. inline constexpr bool is_execution_policy_v<execution::parallel_unsequenced_policy> = true;
  79. template <>
  80. inline constexpr bool is_execution_policy_v<execution::__unsequenced_policy> = true;
  81. template <>
  82. inline constexpr bool __is_parallel_execution_policy_impl<execution::parallel_policy> = true;
  83. template <>
  84. inline constexpr bool __is_parallel_execution_policy_impl<execution::parallel_unsequenced_policy> = true;
  85. template <>
  86. inline constexpr bool __is_unsequenced_execution_policy_impl<execution::__unsequenced_policy> = true;
  87. template <>
  88. inline constexpr bool __is_unsequenced_execution_policy_impl<execution::parallel_unsequenced_policy> = true;
  89. # if _LIBCPP_STD_VER >= 20
  90. template <>
  91. inline constexpr bool is_execution_policy_v<execution::unsequenced_policy> = true;
  92. template <>
  93. inline constexpr bool __is_unsequenced_execution_policy_impl<execution::unsequenced_policy> = true;
  94. # endif
  95. template <class _Tp>
  96. struct is_execution_policy : bool_constant<is_execution_policy_v<_Tp>> {};
  97. template <class _ExecutionPolicy>
  98. _LIBCPP_HIDE_FROM_ABI auto __remove_parallel_policy(const _ExecutionPolicy&) {
  99. if constexpr (is_same_v<_ExecutionPolicy, execution::parallel_policy>) {
  100. return execution::sequenced_policy(execution::__disable_user_instantiations_tag{});
  101. } else if constexpr (is_same_v<_ExecutionPolicy, execution::parallel_unsequenced_policy>) {
  102. return execution::__unsequenced_policy{execution::__disable_user_instantiations_tag{}};
  103. }
  104. }
  105. _LIBCPP_END_NAMESPACE_STD
  106. #endif // !defined(_LIBCPP_HAS_NO_INCOMPLETE_PSTL) && _LIBCPP_STD_VER >= 17
  107. #endif // _LIBCPP_EXECUTION