-
Notifications
You must be signed in to change notification settings - Fork 59
Expand file tree
/
Copy pathmailbox.h
More file actions
231 lines (202 loc) · 8.57 KB
/
mailbox.h
File metadata and controls
231 lines (202 loc) · 8.57 KB
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
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
/*
Copyright (c) 2005-2019 Intel Corporation
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.
*/
#ifndef _TBB_mailbox_H
#define _TBB_mailbox_H
#include "tbb/tbb_stddef.h"
#include "tbb/cache_aligned_allocator.h"
#include "scheduler_common.h"
#include "tbb/atomic.h"
namespace tbb {
namespace internal {
struct task_proxy : public task {
static const intptr_t pool_bit = 1<<0;
static const intptr_t mailbox_bit = 1<<1;
static const intptr_t location_mask = pool_bit | mailbox_bit;
/* All but two low-order bits represent a (task*).
Two low-order bits mean:
1 = proxy is/was/will be in task pool
2 = proxy is/was/will be in mailbox */
intptr_t task_and_tag;
//! Pointer to next task_proxy in a mailbox
task_proxy *__TBB_atomic next_in_mailbox;
//! Mailbox to which this was mailed.
mail_outbox* outbox;
//! True if the proxy is stored both in its sender's pool and in the destination mailbox.
static bool is_shared ( intptr_t tat ) {
return (tat & location_mask) == location_mask;
}
//! Returns a pointer to the encapsulated task or NULL.
static task* task_ptr ( intptr_t tat ) {
return (task*)(tat & ~location_mask);
}
//! Returns a pointer to the encapsulated task or NULL, and frees proxy if necessary.
template<intptr_t from_bit>
inline task* extract_task () {
__TBB_ASSERT( prefix().extra_state == es_task_proxy, "Normal task misinterpreted as a proxy?" );
intptr_t tat = __TBB_load_with_acquire(task_and_tag);
__TBB_ASSERT( tat == from_bit || (is_shared(tat) && task_ptr(tat)),
"Proxy's tag cannot specify both locations if the proxy "
"was retrieved from one of its original locations" );
if ( tat != from_bit ) {
const intptr_t cleaner_bit = location_mask & ~from_bit;
// Attempt to transition the proxy to the "empty" state with
// cleaner_bit specifying entity responsible for its eventual freeing.
// Explicit cast to void* is to work around a seeming ICC 11.1 bug.
if ( as_atomic(task_and_tag).compare_and_swap(cleaner_bit, tat) == tat ) {
// Successfully grabbed the task, and left new owner with the job of freeing the proxy
return task_ptr(tat);
}
}
// Proxied task has already been claimed from another proxy location.
__TBB_ASSERT( task_and_tag == from_bit, "Empty proxy cannot contain non-zero task pointer" );
return NULL;
}
}; // struct task_proxy
//! Internal representation of mail_outbox, without padding.
class unpadded_mail_outbox {
protected:
typedef task_proxy*__TBB_atomic proxy_ptr;
//! Pointer to first task_proxy in mailbox, or NULL if box is empty.
proxy_ptr my_first;
//! Pointer to pointer that will point to next item in the queue. Never NULL.
proxy_ptr* __TBB_atomic my_last;
//! Owner of mailbox is not executing a task, and has drained its own task pool.
bool my_is_idle;
};
//! Class representing where mail is put.
/** Padded to occupy a cache line. */
class mail_outbox : padded<unpadded_mail_outbox> {
task_proxy* internal_pop( __TBB_ISOLATION_EXPR(isolation_tag isolation) ) {
task_proxy* curr = __TBB_load_relaxed( my_first );
if ( !curr )
return NULL;
task_proxy **prev_ptr = &my_first;
#if __TBB_TASK_ISOLATION
if ( isolation != no_isolation ) {
while ( curr->prefix().isolation != isolation ) {
prev_ptr = &curr->next_in_mailbox;
curr = curr->next_in_mailbox;
if ( !curr )
return NULL;
}
}
#endif /* __TBB_TASK_ISOLATION */
__TBB_control_consistency_helper(); // on my_first
// There is a first item in the mailbox. See if there is a second.
if ( task_proxy* second = curr->next_in_mailbox ) {
// There are at least two items, so first item can be popped easily.
*prev_ptr = second;
} else {
// There is only one item. Some care is required to pop it.
*prev_ptr = NULL;
if ( as_atomic( my_last ).compare_and_swap( prev_ptr, &curr->next_in_mailbox ) == &curr->next_in_mailbox ) {
// Successfully transitioned mailbox from having one item to having none.
__TBB_ASSERT( !curr->next_in_mailbox, NULL );
} else {
// Some other thread updated my_last but has not filled in first->next_in_mailbox
// Wait until first item points to second item.
atomic_backoff backoff;
while ( !(second = curr->next_in_mailbox) ) backoff.pause();
*prev_ptr = second;
}
}
__TBB_ASSERT( curr, NULL );
return curr;
}
public:
friend class mail_inbox;
//! Push task_proxy onto the mailbox queue of another thread.
/** Implementation is wait-free. */
void push( task_proxy* t ) {
__TBB_ASSERT(t, NULL);
t->next_in_mailbox = NULL;
proxy_ptr * const link = (proxy_ptr *)__TBB_FetchAndStoreW(&my_last,(intptr_t)&t->next_in_mailbox);
// No release fence required for the next store, because there are no memory operations
// between the previous fully fenced atomic operation and the store.
__TBB_store_relaxed(*link, t);
}
//! Return true if mailbox is empty
bool empty() {
return __TBB_load_relaxed(my_first) == NULL;
}
//! Construct *this as a mailbox from zeroed memory.
/** Raise assertion if *this is not previously zeroed, or sizeof(this) is wrong.
This method is provided instead of a full constructor since we know the object
will be constructed in zeroed memory. */
void construct() {
__TBB_ASSERT( sizeof(*this)==NFS_MaxLineSize, NULL );
__TBB_ASSERT( !my_first, NULL );
__TBB_ASSERT( !my_last, NULL );
__TBB_ASSERT( !my_is_idle, NULL );
my_last=&my_first;
suppress_unused_warning(pad);
}
//! Drain the mailbox
intptr_t drain() {
intptr_t k = 0;
// No fences here because other threads have already quit.
for( ; task_proxy* t = my_first; ++k ) {
my_first = t->next_in_mailbox;
NFS_Free((char*)t - task_prefix_reservation_size);
}
return k;
}
//! True if thread that owns this mailbox is looking for work.
bool recipient_is_idle() {
return my_is_idle;
}
}; // class mail_outbox
//! Class representing source of mail.
class mail_inbox {
//! Corresponding sink where mail that we receive will be put.
mail_outbox* my_putter;
public:
//! Construct unattached inbox
mail_inbox() : my_putter(NULL) {}
//! Attach inbox to a corresponding outbox.
void attach( mail_outbox& putter ) {
my_putter = &putter;
}
//! Detach inbox from its outbox
void detach() {
__TBB_ASSERT(my_putter,"not attached");
my_putter = NULL;
}
//! Get next piece of mail, or NULL if mailbox is empty.
task_proxy* pop( __TBB_ISOLATION_EXPR( isolation_tag isolation ) ) {
return my_putter->internal_pop( __TBB_ISOLATION_EXPR( isolation ) );
}
//! Return true if mailbox is empty
bool empty() {
return my_putter->empty();
}
//! Indicate whether thread that reads this mailbox is idle.
/** Raises assertion failure if mailbox is redundantly marked as not idle. */
void set_is_idle( bool value ) {
if( my_putter ) {
__TBB_ASSERT( my_putter->my_is_idle || value, "attempt to redundantly mark mailbox as not idle" );
my_putter->my_is_idle = value;
}
}
//! Indicate whether thread that reads this mailbox is idle.
bool is_idle_state ( bool value ) const {
return !my_putter || my_putter->my_is_idle == value;
}
#if DO_ITT_NOTIFY
//! Get pointer to corresponding outbox used for ITT_NOTIFY calls.
void* outbox() const {return my_putter;}
#endif /* DO_ITT_NOTIFY */
}; // class mail_inbox
} // namespace internal
} // namespace tbb
#endif /* _TBB_mailbox_H */