-
Notifications
You must be signed in to change notification settings - Fork 4
/
runtime.cpp
142 lines (124 loc) · 3.34 KB
/
runtime.cpp
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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
#include <cstdio>
#include <cstdlib>
#include <sys/time.h>
#include <pthread.h>
#include <unistd.h>
#include <map>
using namespace std;
pthread_mutex_t mutex;
map<int, int> sizes;
void add_size(void *key, int val) {
pthread_mutex_lock(&mutex);
sizes[long(key)] = val;
pthread_mutex_unlock(&mutex);
}
int get_size(void *key) {
pthread_mutex_lock(&mutex);
int size = sizes[long(key)];
pthread_mutex_unlock(&mutex);
return size;
}
extern "C" {
const bool debug = false;
void *hlvm_alloc(int n, int m) {
if (debug)
printf("hlvm_alloc(%d, %d)\n", n, m);
if (n*m == 0) {
if (debug)
printf("hlvm_alloc(%d, %d) -> %p\n", n, m, (void *)NULL);
return 0;
}
//void *data = calloc(n, m);
void *data = malloc(n*m);
if (data == 0) {
printf("Out of memory\n");
exit(1);
}
if (debug)
printf("hlvm_alloc(%d, %d) -> %p\n", n, m, data);
if (debug)
add_size(data, n*m);
return data;
}
void hlvm_free(void *data) {
if (data != 0) {
int size = 0;
if (debug)
size = get_size(data);
if (debug)
sizes.erase(long(data));
free(data);
}
}
pthread_attr_t attr;
pthread_key_t key;
__thread void *local;
void hlvm_init() {
if (debug) {
printf("hlvm_init()\n");
printf("Initializing pthread attribute...\n");
}
pthread_attr_init(&attr);
pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_JOINABLE);
if (debug)
printf("Creating pthread key...\n");
key = pthread_key_create(&key, NULL);
pthread_mutex_init(&mutex, NULL);
if (debug)
printf("hlvm_init() ends\n");
}
void *hlvm_create_thread(void *(*f)(void *), void *x) {
if (debug)
printf("hlvm_create_thread(%p, %p)\n", f, x);
pthread_t *thread = (pthread_t *)hlvm_alloc(1, sizeof(pthread_t));
pthread_create(thread, &attr, f, x);
if (debug)
printf("hlvm_create_thread(%p, %p) -> %p\n", f, x, thread);
return (void *)thread;
}
void hlvm_join_thread(pthread_t *thread) {
if (debug)
printf("hlvm_join_thread(%p)\n", thread);
pthread_join(*thread, NULL);
hlvm_free(thread);
if (debug)
printf("hlvm_join_thread(%p) done\n", thread);
}
void *hlvm_create_mutex() {
if (debug)
printf("hlvm_create_mutex()\n");
pthread_mutex_t *mutex =
(pthread_mutex_t *)hlvm_alloc(1, sizeof(pthread_mutex_t));
pthread_mutex_init(mutex, NULL);
if (debug)
printf("hlvm_create_mutex() -> %p\n", mutex);
return (void *)mutex;
}
void hlvm_lock_mutex(pthread_mutex_t *mutex) {
pthread_mutex_lock(mutex);
}
void hlvm_unlock_mutex(pthread_mutex_t *mutex) {
pthread_mutex_unlock(mutex);
}
void *hlvm_get_thread_local() {
void *tl = pthread_getspecific(key);
//printf("hlvm_get_thread_local() -> %p\n", tl);
return tl;
}
void hlvm_set_thread_local(void *ptr) {
if (debug)
printf("hlvm_set_thread_local(%p)\n", ptr);
pthread_setspecific(key, ptr);
if (debug)
printf("hlvm_set_thread_local(%p) done\n", ptr);
}
// If *ptr is oldval then replace with newval, returning the old *ptr
int hlvm_cas(int *ptr, int oldval, int newval) {
return __sync_val_compare_and_swap(ptr, oldval, newval);
}
double hlvm_time() {
struct timeval t;
gettimeofday(&t, NULL);
return (double)t.tv_sec + 1e-6 * (double)t.tv_usec;
}
}