Paris7/work-stealing-scheduler
Archived
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

not working but still progress :)

Browse source
This commit is contained in:
Mylloon 2024-04-21 16:51:26 +02:00
parent 6ea1c8d0dd
commit 386ec75895
Signed by: Anri
GPG key ID: A82D63DFF8D1317F
1 changed files with 20 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
src/sched.c
View file

@ -49,7 +49,7 @@ int sched_init_cleanup(int);
int sched_spawn_core(taskfunc, void *, struct scheduler *, int); int sched_spawn_core(taskfunc, void *, struct scheduler *, int);
/* Récupère l'index du thread courant */ /* Récupère l'index du thread courant */
int current_thread(void); int current_thread(struct scheduler *);
int int
sched_init(int nthreads, int qlen, taskfunc f, void *closure) sched_init(int nthreads, int qlen, taskfunc f, void *closure)
@ -204,21 +204,32 @@ sched_init_cleanup(int ret_code)
int int
sched_spawn(taskfunc f, void *closure, struct scheduler *s) sched_spawn(taskfunc f, void *closure, struct scheduler *s)
{ {
int core;
if((core = current_thread(s)) < 0) {
fprintf(stderr, "Thread not in list, who am I?\n");
return -1;
}
// On ajoute la tâche sur la pile du thread courant // On ajoute la tâche sur la pile du thread courant
return sched_spawn_core(f, closure, s, current_thread()); return sched_spawn_core(f, closure, s, core);
} }
int int
current_thread() current_thread(struct scheduler *s)
{ {
/* TODO: Récupère le thread courant */ pthread_t current = pthread_self();
return 0; for(int i = 0; i < s->nthreads; i++) {
if(pthread_equal(s->threads[i], current)) {
return i;
}
}
return -1;
} }
int int
sched_spawn_core(taskfunc f, void *closure, struct scheduler *s, int core) sched_spawn_core(taskfunc f, void *closure, struct scheduler *s, int core)
{ {
pthread_mutex_lock(&s->mutex[core]); pthread_mutex_lock(&s->mutex[core]);
if(s->top[core] + 1 >= s->qlen) { if(s->top[core] + 1 >= s->qlen) {
@ -240,11 +251,11 @@ sched_spawn_core(taskfunc f, void *closure, struct scheduler *s, int core)
void * void *
sched_worker(void *arg) sched_worker(void *arg)
{ {
// TODO: Récupère le processus courant (ID = index tableau schedulers)
int curr_th = 0;
struct scheduler *s = (struct scheduler *)arg; struct scheduler *s = (struct scheduler *)arg;
// Récupère le processus courant (index tableau)
int curr_th = current_thread(s);
while(1) { while(1) {
pthread_mutex_lock(&s->mutex[curr_th]); pthread_mutex_lock(&s->mutex[curr_th]);