X-Git-Url: http://demsky.eecs.uci.edu/git/?a=blobdiff_plain;f=lib%2FSystem%2FThreading.cpp;h=3b0bc72eca9e80b796b0b63373d82414e78a97d9;hb=eebe970c36cbc571910587bfc2efbd3bec47a9ff;hp=bc41afee66bafca2c8a261dcd41d0bb3feba7fe3;hpb=3d7622d0f5849fe2c1113c5751d4cfbc169d810f;p=oota-llvm.git diff --git a/lib/System/Threading.cpp b/lib/System/Threading.cpp index bc41afee66b..3b0bc72eca9 100644 --- a/lib/System/Threading.cpp +++ b/lib/System/Threading.cpp @@ -11,12 +11,12 @@ // //===----------------------------------------------------------------------===// -#include "llvm/Config/config.h" #include "llvm/System/Threading.h" #include "llvm/System/Atomic.h" #include "llvm/System/Mutex.h" #include "llvm/Config/config.h" #include + using namespace llvm; static bool multithreaded_mode = false; @@ -62,3 +62,55 @@ void llvm::llvm_acquire_global_lock() { void llvm::llvm_release_global_lock() { if (multithreaded_mode) global_lock->release(); } + +#if defined(LLVM_MULTITHREADED) && defined(HAVE_PTHREAD_H) +#include + +struct ThreadInfo { + void (*UserFn)(void *); + void *UserData; +}; +static void *ExecuteOnThread_Dispatch(void *Arg) { + ThreadInfo *TI = reinterpret_cast(Arg); + TI->UserFn(TI->UserData); + return 0; +} + +void llvm::llvm_execute_on_thread(void (*Fn)(void*), void *UserData, + unsigned RequestedStackSize) { + ThreadInfo Info = { Fn, UserData }; + pthread_attr_t Attr; + pthread_t Thread; + + // Construct the attributes object. + if (::pthread_attr_init(&Attr) != 0) + return; + + // Set the requested stack size, if given. + if (RequestedStackSize != 0) { + if (::pthread_attr_setstacksize(&Attr, RequestedStackSize) != 0) + goto error; + } + + // Construct and execute the thread. + if (::pthread_create(&Thread, &Attr, ExecuteOnThread_Dispatch, &Info) != 0) + goto error; + + // Wait for the thread and clean up. + ::pthread_join(Thread, 0); + + error: + ::pthread_attr_destroy(&Attr); +} + +#else + +// No non-pthread implementation, currently. + +void llvm::llvm_execute_on_thread(void (*Fn)(void*), void *UserData, + unsigned RequestedStackSize) { + (void) RequestedStackSize; + Fn(UserData); +} + +#endif