mirror of
https://github.com/nim-lang/Nim.git
synced 2026-01-03 03:32:32 +00:00
hopefully faster barrier implementation
This commit is contained in:
@@ -42,22 +42,30 @@ proc signal(cv: var CondVar) =
|
||||
|
||||
type
|
||||
Barrier* {.compilerProc.} = object
|
||||
counter: int
|
||||
entered: int
|
||||
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.} =
|
||||
atomicInc b.counter
|
||||
atomicInc b.entered
|
||||
|
||||
proc barrierLeave*(b: ptr Barrier) {.compilerProc.} =
|
||||
atomicDec b.counter
|
||||
if b.counter <= 0: signal(b.cv)
|
||||
atomicInc b.left
|
||||
# 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.} =
|
||||
b.counter = 0
|
||||
b.entered = 0
|
||||
b.cv = createCondVar()
|
||||
b.left = -1
|
||||
|
||||
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)
|
||||
|
||||
{.pop.}
|
||||
|
||||
Reference in New Issue
Block a user