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
|
#include <algorithm>
#include <cassert>
#include <chrono>
#include <cstddef>
#include <cstdint>
#include <cstdlib>
#include <iostream>
#include <memory>
#include <numeric>
#include <stdexcept>
#include <vector>
extern "C" {
#include <pthread.h>
#include <signal.h>
}
#include "buffered_channel.hpp"
using channel_type = buffered_channel< std::uint64_t >;
using clock_type = std::chrono::steady_clock;
using duration_type = clock_type::duration;
using time_point_type = clock_type::time_point;
struct thread_args {
channel_type & c;
std::size_t num;
std::size_t size;
std::size_t div;
};
// microbenchmark
void * skynet( void * vargs) {
thread_args * args = static_cast< thread_args * >( vargs);
if ( 1 == args->size) {
args->c.push( args->num);
} else {
channel_type rc{ 16 };
for ( std::size_t i = 0; i < args->div; ++i) {
auto sub_num = args->num + i * args->size / args->div;
auto sub_size = args->size / args->div;
auto size = 8 * 1024;
pthread_attr_t tattr;
if ( 0 != ::pthread_attr_init( & tattr) ) {
std::runtime_error("pthread_attr_init() failed");
}
if ( 0 != ::pthread_attr_setstacksize( & tattr, size) ) {
std::runtime_error("pthread_attr_setstacksize() failed");
}
thread_args * targs = new thread_args{ rc, sub_num, sub_size, args->div };
pthread_t tid;
if ( 0 != ::pthread_create( & tid, & tattr, & skynet, targs) ) {
std::runtime_error("pthread_create() failed");
}
if ( 0 != ::pthread_detach( tid) ) {
std::runtime_error("pthread_detach() failed");
}
}
std::uint64_t sum{ 0 };
for ( std::size_t i = 0; i < args->div; ++i) {
sum += rc.value_pop();
}
args->c.push( sum);
}
delete args;
return nullptr;
}
int main() {
try {
std::size_t size{ 10000 };
std::size_t div{ 10 };
std::uint64_t result{ 0 };
duration_type duration{ duration_type::zero() };
channel_type rc{ 2 };
thread_args * targs = new thread_args{ rc, 0, size, div };
time_point_type start{ clock_type::now() };
skynet( targs);
result = rc.value_pop();
duration = clock_type::now() - start;
std::cout << "Result: " << result << " in " << duration.count() / 1000000 << " ms" << std::endl;
std::cout << "done." << std::endl;
return EXIT_SUCCESS;
} catch ( std::exception const& e) {
std::cerr << "exception: " << e.what() << std::endl;
} catch (...) {
std::cerr << "unhandled exception" << std::endl;
}
return EXIT_FAILURE;
}
|
