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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
// 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
// ----------------------------------------------------------------------------
/*
{xrst_begin team_openmp.cpp}
OpenMP Implementation of a Team of AD Threads
#############################################
See :ref:`team_thread.hpp-name` for this routines specifications.
{xrst_literal
// BEGIN C++
// END C++
}
{xrst_end team_openmp.cpp}
*/
// BEGIN C++
# include <omp.h>
# include <cppad/cppad.hpp>
# include "../team_thread.hpp"
namespace {
using CppAD::thread_alloc;
// number of threads in this team
size_t num_threads_;
// used to inform CppAD when we are in parallel execution mode
bool in_parallel(void)
{ return omp_in_parallel() != 0; }
// used to inform CppAD of the current thread number
size_t thread_num(void)
{ return static_cast<size_t>( omp_get_thread_num() ); }
}
bool team_create(size_t num_threads)
{
bool ok = ! in_parallel();
ok &= thread_num() == 0;;
ok &= num_threads > 0;
// Turn off dynamic thread adjustment
omp_set_dynamic(0);
// Set the number of OpenMP threads
omp_set_num_threads( int(num_threads) );
// setup for using CppAD::AD<double> in parallel
thread_alloc::parallel_setup(num_threads, in_parallel, thread_num);
thread_alloc::hold_memory(true);
CppAD::parallel_ad<double>();
// inform team_work of number of threads
num_threads_ = num_threads;
return ok;
}
bool team_work(void worker(void))
{ bool ok = ! in_parallel();
ok &= thread_num() == 0;;
ok &= num_threads_ > 0;
int number_threads = int(num_threads_);
int thread_num;
# pragma omp parallel for
for(thread_num = 0; thread_num < number_threads; thread_num++)
worker();
// end omp parallel for
return ok;
}
bool team_destroy(void)
{ bool ok = ! in_parallel();
ok &= thread_num() == 0;;
ok &= num_threads_ > 0;
// inform team_work of number of threads
num_threads_ = 1;
// Set the number of OpenMP threads to one
omp_set_num_threads( int(num_threads_) );
// inform CppAD no longer in multi-threading mode
thread_alloc::parallel_setup(num_threads_, nullptr, nullptr);
thread_alloc::hold_memory(false);
CppAD::parallel_ad<double>();
return ok;
}
const char* team_name(void)
{ return "openmp"; }
// END C++
|
