commit | 0a4b81c950fa4fb08037ceeaa54d008efdf4e739 | [log] [tgz] |
---|---|---|
author | Marat Dukhan <maratek@google.com> | Tue Aug 22 08:52:05 2023 -0700 |
committer | Marat Dukhan <maratek@gmail.com> | Tue Aug 22 09:52:51 2023 -0600 |
tree | ea50df11aaabecc0ed7e2b9fa64a8389548eac0f | |
parent | 32f2a933ae830d515b05a2635e8dc56bfe221329 [diff] |
Fix pthreadpool_parallelize_*_with_thread functions When work-stealing, pass the thread ID of the executing thread rather than of the thread we steal the work from
pthreadpool is a portable and efficient thread pool implementation. It provides similar functionality to #pragma omp parallel for
, but with additional features.
The following example demonstates using the thread pool for parallel addition of two arrays:
static void add_arrays(struct array_addition_context* context, size_t i) { context->sum[i] = context->augend[i] + context->addend[i]; } #define ARRAY_SIZE 4 int main() { double augend[ARRAY_SIZE] = { 1.0, 2.0, 4.0, -5.0 }; double addend[ARRAY_SIZE] = { 0.25, -1.75, 0.0, 0.5 }; double sum[ARRAY_SIZE]; pthreadpool_t threadpool = pthreadpool_create(0); assert(threadpool != NULL); const size_t threads_count = pthreadpool_get_threads_count(threadpool); printf("Created thread pool with %zu threads\n", threads_count); struct array_addition_context context = { augend, addend, sum }; pthreadpool_parallelize_1d(threadpool, (pthreadpool_task_1d_t) add_arrays, (void*) &context, ARRAY_SIZE, PTHREADPOOL_FLAG_DISABLE_DENORMALS /* flags */); pthreadpool_destroy(threadpool); threadpool = NULL; printf("%8s\t%.2lf\t%.2lf\t%.2lf\t%.2lf\n", "Augend", augend[0], augend[1], augend[2], augend[3]); printf("%8s\t%.2lf\t%.2lf\t%.2lf\t%.2lf\n", "Addend", addend[0], addend[1], addend[2], addend[3]); printf("%8s\t%.2lf\t%.2lf\t%.2lf\t%.2lf\n", "Sum", sum[0], sum[1], sum[2], sum[3]); return 0; }