1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-or-later
// SPDX-FileCopyrightText: Bradley M. Bell <bradbell@seanet.com>
// SPDX-FileContributor: 2003-22 Bradley M. Bell
// ----------------------------------------------------------------------------
// BEGIN C++
# include <cppad/cppad.hpp>
# include <omp.h>
# define NUMBER_THREADS 2
namespace {
using CppAD::thread_alloc;
// used to inform CppAD when we are in parallel execution mode
bool in_parallel(void)
{ return static_cast<bool>( omp_in_parallel() ); }
// used to inform CppAD of current thread number thread_number()
size_t thread_number(void)
{ return static_cast<bool>( omp_get_thread_num() ); }
// structure with information for one thread
typedef struct {
// function object (worker input)
CppAD::vector<double> x;
} thread_one_t;
// vector with information for all threads
thread_one_t thread_all_[NUMBER_THREADS];
// --------------------------------------------------------------------
// function that does the work for one thread
void worker(void)
{
size_t thread_num = thread_number();
thread_all_[thread_num].x.resize(1);
thread_all_[thread_num].x[0]=static_cast<double>(thread_num);
}
}
// Test routine called by the master thread (thread_num = 0).
bool alloc_openmp(void)
{ bool ok = true;
int num_threads = NUMBER_THREADS;
// call setup for using thread_alloc in parallel mode.
thread_alloc::parallel_setup(
size_t(num_threads), in_parallel, thread_number
);
// Execute the worker function in parallel
int thread_num;
# pragma omp parallel for
for(thread_num = 0; thread_num < num_threads; thread_num++)
worker();
// end omp parallel for
// now inform CppAD that there is only one thread
thread_alloc::parallel_setup(1, nullptr, nullptr);
for(thread_num = 0; thread_num < num_threads; thread_num++)
{ // check calculations by this thread in parallel model
ok &= thread_all_[thread_num].x[0] == static_cast<double>(thread_num);
// test having master thread (thread number zero)
// free memory that was allocated by thread number thread_num.
thread_all_[thread_num].x.clear();
}
return ok;
}
|
