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
99
100
101
102
103
104
105
106
107
108
109
110
|
/* -------------------------------------------------------------------------- *
* Simbody(tm): SimTKcommon *
* -------------------------------------------------------------------------- *
* This is part of the SimTK biosimulation toolkit originating from *
* Simbios, the NIH National Center for Physics-Based Simulation of *
* Biological Structures at Stanford, funded under the NIH Roadmap for *
* Medical Research, grant U54 GM072970. See https://simtk.org/home/simbody. *
* *
* Portions copyright (c) 2008-12 Stanford University and the Authors. *
* Authors: Peter Eastman *
* Contributors: *
* *
* Licensed under the Apache License, Version 2.0 (the "License"); you may *
* not use this file except in compliance with the License. You may obtain a *
* copy of the License at http://www.apache.org/licenses/LICENSE-2.0. *
* *
* Unless required by applicable law or agreed to in writing, software *
* distributed under the License is distributed on an "AS IS" BASIS, *
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. *
* See the License for the specific language governing permissions and *
* limitations under the License. *
* -------------------------------------------------------------------------- */
#include "SimTKcommon.h"
#include <iostream>
#define ASSERT(cond) {SimTK_ASSERT_ALWAYS(cond, "Assertion failed");}
using std::cout;
using std::endl;
using namespace SimTK;
using namespace std;
bool isParallel;
class SetFlagTask : public ParallelExecutor::Task {
public:
SetFlagTask(Array_<int>& flags, int& count) : flags(flags), count(count) {
}
void execute(int index) override {
flags[index]++;
localCount++;
ASSERT(ParallelExecutor::isWorkerThread() == isParallel);
}
void initialize() override {
localCount = 0;
ASSERT(ParallelExecutor::isWorkerThread() == isParallel);
}
void finish() override {
count += localCount;
ASSERT(ParallelExecutor::isWorkerThread() == isParallel);
}
private:
Array_<int>& flags;
int& count;
static thread_local int localCount;
};
/*static*/ thread_local int SetFlagTask::localCount = 0;
void testParallelExecution() {
const int numFlags = 100;
Array_<int> flags(numFlags);
isParallel = (ParallelExecutor::getNumProcessors() > 1);
ParallelExecutor executor;
ASSERT(!ParallelExecutor::isWorkerThread());
for (int i = 0; i < 100; ++i) {
int count = 0;
SetFlagTask task(flags, count);
for (int j = 0; j < numFlags; ++j)
flags[j] = 0;
executor.execute(task, numFlags-10);
ASSERT(count == numFlags-10);
for (int j = 0; j < numFlags; ++j)
ASSERT(flags[j] == (j < numFlags-10 ? 1 : 0));
}
ASSERT(!ParallelExecutor::isWorkerThread());
}
void testSingleThreadedExecution() {
const int numFlags = 100;
Array_<int> flags(numFlags);
isParallel = false;
ParallelExecutor executor(1); // Specify only a single thread.
int count = 0;
SetFlagTask task(flags, count);
for (int j = 0; j < numFlags; ++j)
flags[j] = 0;
executor.execute(task, numFlags-10);
ASSERT(count == numFlags-10);
for (int j = 0; j < numFlags; ++j)
ASSERT(flags[j] == (j < numFlags-10 ? 1 : 0));
}
void testResizeThreads() {
for(int x = 1; x < 100; ++x)
{
ParallelExecutor executor(x);
SimTK_TEST(executor.getMaxThreads() == x);
}
}
int main() {
SimTK_START_TEST("TestParallelExecutor");
SimTK_SUBTEST(testParallelExecution);
SimTK_SUBTEST(testSingleThreadedExecution);
SimTK_SUBTEST(testResizeThreads);
SimTK_END_TEST();
return 0;
}
|
