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
|
// -*- mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*-
// vi: set et ts=4 sw=4 sts=4:
/*
This file is part of the Open Porous Media project (OPM).
OPM is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 2 of the License, or
(at your option) any later version.
OPM is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with OPM. If not, see <http://www.gnu.org/licenses/>.
Consult the COPYING file in the top-level source directory of this
module for the precise wording of the license and the list of
copyright holders.
*/
/*!
* \file
*
* \brief This file serves as an example of how to use the tasklet mechanism for
* asynchronous work.
*/
#include "config.h"
#include <opm/models/parallel/tasklets.hpp>
#include <cassert>
#include <chrono>
#include <iostream>
std::mutex outputMutex;
// The runner is created on the heap for the assertion and outputs in the run function of the tasklets.
std::unique_ptr<Opm::TaskletRunner> runner{};
class SleepTasklet : public Opm::TaskletInterface
{
public:
explicit SleepTasklet(int mseconds)
: mseconds_(mseconds)
{
n_ = numInstantiated_;
++ numInstantiated_;
}
void run() override
{
assert(0 <= runner->workerThreadIndex() && runner->workerThreadIndex() < runner->numWorkerThreads());
std::this_thread::sleep_for(std::chrono::milliseconds(mseconds_));
std::lock_guard<std::mutex> guard(outputMutex);
std::cout << "Sleep tasklet " << n_ << " of " << mseconds_ << " ms completed by worker thread " << runner->workerThreadIndex() << std::endl;
}
private:
static int numInstantiated_;
int n_;
int mseconds_;
};
void sleepAndPrintFunction();
void sleepAndPrintFunction()
{
int ms = 100;
std::this_thread::sleep_for(std::chrono::milliseconds(ms));
std::lock_guard<std::mutex> guard(outputMutex);
std::cout << "Sleep completed by worker thread " << runner->workerThreadIndex() << std::endl;
}
int SleepTasklet::numInstantiated_ = 0;
int main()
{
int numWorkers = 2;
runner = std::make_unique<Opm::TaskletRunner>(numWorkers);
// the master thread is not a worker thread
assert(runner->workerThreadIndex() < 0);
assert(runner->numWorkerThreads() == numWorkers);
for (int i = 0; i < 5; ++ i) {
//auto st = std::make_shared<SleepTasklet>((i + 1)*1000);
auto st = std::make_shared<SleepTasklet>(100);
runner->dispatch(st);
}
std::cout << "before barrier" << std::endl;
runner->barrier();
std::cout << "after barrier" << std::endl;
runner->dispatchFunction(sleepAndPrintFunction);
runner->dispatchFunction(sleepAndPrintFunction, /*numInvokations=*/6);
return 0;
}
|
