Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line |
|---|---|---|---|
| 14 | pmbaty | 1 | //===-- llvm/Support/thread.h - Wrapper for <thread> ------------*- C++ -*-===// |
| 2 | // |
||
| 3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
||
| 4 | // See https://llvm.org/LICENSE.txt for license information. |
||
| 5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
||
| 6 | // |
||
| 7 | //===----------------------------------------------------------------------===// |
||
| 8 | // |
||
| 9 | // This header is a wrapper for <thread> that works around problems with the |
||
| 10 | // MSVC headers when exceptions are disabled. It also provides llvm::thread, |
||
| 11 | // which is either a typedef of std::thread or a replacement that calls the |
||
| 12 | // function synchronously depending on the value of LLVM_ENABLE_THREADS. |
||
| 13 | // |
||
| 14 | //===----------------------------------------------------------------------===// |
||
| 15 | |||
| 16 | #ifndef LLVM_SUPPORT_THREAD_H |
||
| 17 | #define LLVM_SUPPORT_THREAD_H |
||
| 18 | |||
| 19 | #include "llvm/Config/llvm-config.h" |
||
| 20 | #include <optional> |
||
| 21 | |||
| 22 | #ifdef _WIN32 |
||
| 23 | typedef unsigned long DWORD; |
||
| 24 | typedef void *PVOID; |
||
| 25 | typedef PVOID HANDLE; |
||
| 26 | #endif |
||
| 27 | |||
| 28 | #if LLVM_ENABLE_THREADS |
||
| 29 | |||
| 30 | #include <thread> |
||
| 31 | |||
| 32 | namespace llvm { |
||
| 33 | |||
| 34 | #if LLVM_ON_UNIX || _WIN32 |
||
| 35 | |||
| 36 | /// LLVM thread following std::thread interface with added constructor to |
||
| 37 | /// specify stack size. |
||
| 38 | class thread { |
||
| 39 | template <typename CalleeTuple> static void GenericThreadProxy(void *Ptr) { |
||
| 40 | std::unique_ptr<CalleeTuple> Callee(static_cast<CalleeTuple *>(Ptr)); |
||
| 41 | std::apply( |
||
| 42 | [](auto &&F, auto &&...Args) { |
||
| 43 | std::forward<decltype(F)>(F)(std::forward<decltype(Args)>(Args)...); |
||
| 44 | }, |
||
| 45 | *Callee); |
||
| 46 | } |
||
| 47 | |||
| 48 | public: |
||
| 49 | #if LLVM_ON_UNIX |
||
| 50 | using native_handle_type = pthread_t; |
||
| 51 | using id = pthread_t; |
||
| 52 | using start_routine_type = void *(*)(void *); |
||
| 53 | |||
| 54 | template <typename CalleeTuple> static void *ThreadProxy(void *Ptr) { |
||
| 55 | GenericThreadProxy<CalleeTuple>(Ptr); |
||
| 56 | return nullptr; |
||
| 57 | } |
||
| 58 | #elif _WIN32 |
||
| 59 | using native_handle_type = HANDLE; |
||
| 60 | using id = DWORD; |
||
| 61 | using start_routine_type = unsigned(__stdcall *)(void *); |
||
| 62 | |||
| 63 | template <typename CalleeTuple> |
||
| 64 | static unsigned __stdcall ThreadProxy(void *Ptr) { |
||
| 65 | GenericThreadProxy<CalleeTuple>(Ptr); |
||
| 66 | return 0; |
||
| 67 | } |
||
| 68 | #endif |
||
| 69 | |||
| 70 | static const std::optional<unsigned> DefaultStackSize; |
||
| 71 | |||
| 72 | thread() : Thread(native_handle_type()) {} |
||
| 73 | thread(thread &&Other) noexcept |
||
| 74 | : Thread(std::exchange(Other.Thread, native_handle_type())) {} |
||
| 75 | |||
| 76 | template <class Function, class... Args> |
||
| 77 | explicit thread(Function &&f, Args &&...args) |
||
| 78 | : thread(DefaultStackSize, f, args...) {} |
||
| 79 | |||
| 80 | template <class Function, class... Args> |
||
| 81 | explicit thread(std::optional<unsigned> StackSizeInBytes, Function &&f, |
||
| 82 | Args &&...args); |
||
| 83 | thread(const thread &) = delete; |
||
| 84 | |||
| 85 | ~thread() { |
||
| 86 | if (joinable()) |
||
| 87 | std::terminate(); |
||
| 88 | } |
||
| 89 | |||
| 90 | thread &operator=(thread &&Other) noexcept { |
||
| 91 | if (joinable()) |
||
| 92 | std::terminate(); |
||
| 93 | Thread = std::exchange(Other.Thread, native_handle_type()); |
||
| 94 | return *this; |
||
| 95 | } |
||
| 96 | |||
| 97 | bool joinable() const noexcept { return Thread != native_handle_type(); } |
||
| 98 | |||
| 99 | inline id get_id() const noexcept; |
||
| 100 | |||
| 101 | native_handle_type native_handle() const noexcept { return Thread; } |
||
| 102 | |||
| 103 | static unsigned hardware_concurrency() { |
||
| 104 | return std::thread::hardware_concurrency(); |
||
| 105 | }; |
||
| 106 | |||
| 107 | inline void join(); |
||
| 108 | inline void detach(); |
||
| 109 | |||
| 110 | void swap(llvm::thread &Other) noexcept { std::swap(Thread, Other.Thread); } |
||
| 111 | |||
| 112 | private: |
||
| 113 | native_handle_type Thread; |
||
| 114 | }; |
||
| 115 | |||
| 116 | thread::native_handle_type |
||
| 117 | llvm_execute_on_thread_impl(thread::start_routine_type ThreadFunc, void *Arg, |
||
| 118 | std::optional<unsigned> StackSizeInBytes); |
||
| 119 | void llvm_thread_join_impl(thread::native_handle_type Thread); |
||
| 120 | void llvm_thread_detach_impl(thread::native_handle_type Thread); |
||
| 121 | thread::id llvm_thread_get_id_impl(thread::native_handle_type Thread); |
||
| 122 | thread::id llvm_thread_get_current_id_impl(); |
||
| 123 | |||
| 124 | template <class Function, class... Args> |
||
| 125 | thread::thread(std::optional<unsigned> StackSizeInBytes, Function &&f, |
||
| 126 | Args &&...args) { |
||
| 127 | typedef std::tuple<std::decay_t<Function>, std::decay_t<Args>...> CalleeTuple; |
||
| 128 | std::unique_ptr<CalleeTuple> Callee( |
||
| 129 | new CalleeTuple(std::forward<Function>(f), std::forward<Args>(args)...)); |
||
| 130 | |||
| 131 | Thread = llvm_execute_on_thread_impl(ThreadProxy<CalleeTuple>, Callee.get(), |
||
| 132 | StackSizeInBytes); |
||
| 133 | if (Thread != native_handle_type()) |
||
| 134 | Callee.release(); |
||
| 135 | } |
||
| 136 | |||
| 137 | thread::id thread::get_id() const noexcept { |
||
| 138 | return llvm_thread_get_id_impl(Thread); |
||
| 139 | } |
||
| 140 | |||
| 141 | void thread::join() { |
||
| 142 | llvm_thread_join_impl(Thread); |
||
| 143 | Thread = native_handle_type(); |
||
| 144 | } |
||
| 145 | |||
| 146 | void thread::detach() { |
||
| 147 | llvm_thread_detach_impl(Thread); |
||
| 148 | Thread = native_handle_type(); |
||
| 149 | } |
||
| 150 | |||
| 151 | namespace this_thread { |
||
| 152 | inline thread::id get_id() { return llvm_thread_get_current_id_impl(); } |
||
| 153 | } // namespace this_thread |
||
| 154 | |||
| 155 | #else // !LLVM_ON_UNIX && !_WIN32 |
||
| 156 | |||
| 157 | /// std::thread backed implementation of llvm::thread interface that ignores the |
||
| 158 | /// stack size request. |
||
| 159 | class thread { |
||
| 160 | public: |
||
| 161 | using native_handle_type = std::thread::native_handle_type; |
||
| 162 | using id = std::thread::id; |
||
| 163 | |||
| 164 | thread() : Thread(std::thread()) {} |
||
| 165 | thread(thread &&Other) noexcept |
||
| 166 | : Thread(std::exchange(Other.Thread, std::thread())) {} |
||
| 167 | |||
| 168 | template <class Function, class... Args> |
||
| 169 | explicit thread(std::optional<unsigned> StackSizeInBytes, Function &&f, |
||
| 170 | Args &&...args) |
||
| 171 | : Thread(std::forward<Function>(f), std::forward<Args>(args)...) {} |
||
| 172 | |||
| 173 | template <class Function, class... Args> |
||
| 174 | explicit thread(Function &&f, Args &&...args) : Thread(f, args...) {} |
||
| 175 | |||
| 176 | thread(const thread &) = delete; |
||
| 177 | |||
| 178 | ~thread() {} |
||
| 179 | |||
| 180 | thread &operator=(thread &&Other) noexcept { |
||
| 181 | Thread = std::exchange(Other.Thread, std::thread()); |
||
| 182 | return *this; |
||
| 183 | } |
||
| 184 | |||
| 185 | bool joinable() const noexcept { return Thread.joinable(); } |
||
| 186 | |||
| 187 | id get_id() const noexcept { return Thread.get_id(); } |
||
| 188 | |||
| 189 | native_handle_type native_handle() noexcept { return Thread.native_handle(); } |
||
| 190 | |||
| 191 | static unsigned hardware_concurrency() { |
||
| 192 | return std::thread::hardware_concurrency(); |
||
| 193 | }; |
||
| 194 | |||
| 195 | inline void join() { Thread.join(); } |
||
| 196 | inline void detach() { Thread.detach(); } |
||
| 197 | |||
| 198 | void swap(llvm::thread &Other) noexcept { std::swap(Thread, Other.Thread); } |
||
| 199 | |||
| 200 | private: |
||
| 201 | std::thread Thread; |
||
| 202 | }; |
||
| 203 | |||
| 204 | namespace this_thread { |
||
| 205 | inline thread::id get_id() { return std::this_thread::get_id(); } |
||
| 206 | } |
||
| 207 | |||
| 208 | #endif // LLVM_ON_UNIX || _WIN32 |
||
| 209 | |||
| 210 | } // namespace llvm |
||
| 211 | |||
| 212 | #else // !LLVM_ENABLE_THREADS |
||
| 213 | |||
| 214 | #include <utility> |
||
| 215 | |||
| 216 | namespace llvm { |
||
| 217 | |||
| 218 | struct thread { |
||
| 219 | thread() {} |
||
| 220 | thread(thread &&other) {} |
||
| 221 | template <class Function, class... Args> |
||
| 222 | explicit thread(std::optional<unsigned> StackSizeInBytes, Function &&f, |
||
| 223 | Args &&...args) { |
||
| 224 | f(std::forward<Args>(args)...); |
||
| 225 | } |
||
| 226 | template <class Function, class... Args> |
||
| 227 | explicit thread(Function &&f, Args &&...args) { |
||
| 228 | f(std::forward<Args>(args)...); |
||
| 229 | } |
||
| 230 | thread(const thread &) = delete; |
||
| 231 | |||
| 232 | void detach() { |
||
| 233 | report_fatal_error("Detaching from a thread does not make sense with no " |
||
| 234 | "threading support"); |
||
| 235 | } |
||
| 236 | void join() {} |
||
| 237 | static unsigned hardware_concurrency() { return 1; }; |
||
| 238 | }; |
||
| 239 | |||
| 240 | } // namespace llvm |
||
| 241 | |||
| 242 | #endif // LLVM_ENABLE_THREADS |
||
| 243 | |||
| 244 | #endif // LLVM_SUPPORT_THREAD_H |