hopefully faster barrier implementation
This commit is contained in:
parent
b78173788d
commit
fbadbb399e
1 changed files with 14 additions and 6 deletions
|
|
@ -42,22 +42,30 @@ proc signal(cv: var CondVar) =
|
||||||
|
|
||||||
type
|
type
|
||||||
Barrier* {.compilerProc.} = object
|
Barrier* {.compilerProc.} = object
|
||||||
counter: int
|
entered: int
|
||||||
cv: CondVar
|
cv: CondVar
|
||||||
|
cacheAlign: array[0..20, byte] # ensure 'left' is not on the same
|
||||||
|
# cache line as 'entered'
|
||||||
|
left: int
|
||||||
|
|
||||||
proc barrierEnter*(b: ptr Barrier) {.compilerProc.} =
|
proc barrierEnter*(b: ptr Barrier) {.compilerProc.} =
|
||||||
atomicInc b.counter
|
atomicInc b.entered
|
||||||
|
|
||||||
proc barrierLeave*(b: ptr Barrier) {.compilerProc.} =
|
proc barrierLeave*(b: ptr Barrier) {.compilerProc.} =
|
||||||
atomicDec b.counter
|
atomicInc b.left
|
||||||
if b.counter <= 0: signal(b.cv)
|
# these can only be equal if 'closeBarrier' already signaled its interest
|
||||||
|
# in this event:
|
||||||
|
if b.left == b.entered: signal(b.cv)
|
||||||
|
|
||||||
proc openBarrier*(b: ptr Barrier) {.compilerProc.} =
|
proc openBarrier*(b: ptr Barrier) {.compilerProc.} =
|
||||||
b.counter = 0
|
b.entered = 0
|
||||||
b.cv = createCondVar()
|
b.cv = createCondVar()
|
||||||
|
b.left = -1
|
||||||
|
|
||||||
proc closeBarrier*(b: ptr Barrier) {.compilerProc.} =
|
proc closeBarrier*(b: ptr Barrier) {.compilerProc.} =
|
||||||
while b.counter > 0: await(b.cv)
|
# signal interest in the "all done" event:
|
||||||
|
atomicInc b.left
|
||||||
|
while b.left != b.entered: await(b.cv)
|
||||||
destroyCondVar(b.cv)
|
destroyCondVar(b.cv)
|
||||||
|
|
||||||
{.pop.}
|
{.pop.}
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue