diff --git a/nx/include/switch.h b/nx/include/switch.h index dd22df23..6c5d8423 100644 --- a/nx/include/switch.h +++ b/nx/include/switch.h @@ -34,6 +34,7 @@ extern "C" { #include "switch/kernel/random.h" #include "switch/kernel/jit.h" #include "switch/kernel/ipc.h" +#include "switch/kernel/barrier.h" #include "switch/services/sm.h" #include "switch/services/smm.h" diff --git a/nx/include/switch/kernel/barrier.h b/nx/include/switch/kernel/barrier.h new file mode 100644 index 00000000..76d2fcd9 --- /dev/null +++ b/nx/include/switch/kernel/barrier.h @@ -0,0 +1,16 @@ +#pragma once +#include "semaphore.h" + + +// barrier structure +typedef struct barrier{ + u64 count; + u64 thread_total; + Semaphore throttle; + Semaphore lock; + Semaphore thread_wait; +} barrier; + +void barrierInit(barrier *my_barrier, u64 thread_count); + +void barrierWait(barrier *b); diff --git a/nx/include/switch/kernel/semaphore.h b/nx/include/switch/kernel/semaphore.h index 12f4c5b0..fac4aa66 100644 --- a/nx/include/switch/kernel/semaphore.h +++ b/nx/include/switch/kernel/semaphore.h @@ -42,3 +42,5 @@ void semaphoreWait(Semaphore *s); * @return true if no wait and successful lock, false otherwise. */ bool semaphoreTryWait(Semaphore *s); + + diff --git a/nx/source/kernel/barrier.c b/nx/source/kernel/barrier.c new file mode 100644 index 00000000..1b248b73 --- /dev/null +++ b/nx/source/kernel/barrier.c @@ -0,0 +1,29 @@ +#include "kernel/barrier.h" + +void barrierInit(barrier *my_barrier, u64 thread_count){ + my_barrier->count = 0; + my_barrier->thread_total = thread_count; + semaphoreInit(&my_barrier->throttle, 0); + semaphoreInit(&my_barrier->lock, 1); + semaphoreInit(&my_barrier->thread_wait, 0); +} + +void barrierWait(barrier *b){ + semaphoreWait(&b->lock); + if(b->count < b->thread_total){ + b->count++; + } + if(b->count < b->thread_total){ + semaphoreSignal(&b->lock); + semaphoreWait(&b->thread_wait); + semaphoreSignal(&b->throttle); + } + else if(b->count == b->thread_total){ + for(int i = 0; i < b->thread_total-1; i++){ + semaphoreSignal(&b->thread_wait); + semaphoreWait(&b->throttle); + } + b->count = 0; + semaphoreSignal(&b->lock); + } +}