status-im
/
nim-taskpools
mirror of https://github.com/status-im/nim-taskpools.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

Re-expose tp.numThreads (#13) 

Necessary for nim-blscurve batch sizing: https://github.com/status-im/nim-blscurve/blob/dad2041/blscurve/bls_batch_verifier.nim#L250
This commit is contained in:
Mamy Ratsimbazafy 2022-02-08 11:55:31 +01:00 committed by GitHub
parent b31b891f11
commit 0a76551afa
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
taskpools/taskpools.nim
View File

@ -90,7 +90,6 @@ type
# Thefts
rng: Rand # RNG state to select victims
numThreads: int
otherDeques: ptr UncheckedArray[ChaseLevDeque[TaskNode]]
victims: SparseSet
@ -102,7 +101,7 @@ type
eventNotifier: EventNotifier
## Puts thread to sleep
numThreads{.align: 64.}: int
numThreads*{.align: 64.}: int
workerDeques: ptr UncheckedArray[ChaseLevDeque[TaskNode]]
## Direct access for task stealing
workers: ptr UncheckedArray[Thread[(Taskpool, WorkerID)]]
@ -127,7 +126,6 @@ proc setupWorker() =
# Thefts
ctx.rng = initRand(0xEFFACED + ctx.id)
ctx.numThreads = ctx.taskpool.numThreads
ctx.otherDeques = ctx.taskpool.workerDeques
ctx.victims.allocate(ctx.taskpool.numThreads)
@ -322,7 +320,7 @@ proc syncAll*(tp: Taskpool) {.raises: [Exception].} =
debug: log("Worker %2d: syncAll 1 - running task 0x%.08x (parent 0x%.08x, current 0x%.08x)\n", ctx.id, taskNode, taskNode.parent, ctx.currentTask)
taskNode.runTask()
if ctx.numThreads == 1 or foreignThreadsParked:
if tp.numThreads == 1 or foreignThreadsParked:
break
# 2. Help other threads