blob: 15f5bbf2f44eeca414d0cb663c0c90c7ba8b6ed4 (
plain)
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
|
/*
* compile with: gcc -Wall -fgnu-tm -O2 -s stm.c -o stm
*/
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <pthread.h>
#include <immintrin.h>
#include <assert.h>
struct thread_data {
pthread_t thrd;
int id;
int result;
};
#define NTHREADS 8
static const unsigned int nthreads = NTHREADS;
static const unsigned int iterations = 4000;
static struct thread_data thrds[NTHREADS];
/* r/w by threads */
static int sum = 0;
void *
thread_func(void *arg)
{
int i;
struct thread_data *td = (struct thread_data *) arg;
printf("Thread %d started ..\n", td->id);
/* critical section - stm */
for (i = 0; i < iterations; i++) {
__transaction_atomic { sum++; }
}
return NULL;
}
int
main(int argc, char *argv[])
{
int i;
for (i = 0; i < nthreads; i++) {
thrds[i].id = i;
thrds[i].result = 0;
assert( pthread_create( &(thrds[i].thrd), NULL, thread_func, (void*)&thrds[i] ) == 0 );
}
for (i = 0; i < nthreads; i++) {
assert( pthread_join(thrds[i].thrd, NULL) == 0 );
}
printf("Expected result: sum == %d\n", iterations*nthreads);
if (sum == iterations*nthreads) {
printf("STM worked! sum == %d\n", sum);
} else {
printf("Wrong result? Maybe your GNU-STM does not work. (sum == %d)\n", sum);
}
return 0;
}
|