66 lines
1.2 KiB
C
66 lines
1.2 KiB
C
|
/* Copyright (c) 2023 Barcelona Supercomputing Center (BSC)
|
||
|
|
* SPDX-License-Identifier: GPL-3.0-or-later */
|
||
|
|
|
||
|
|
#include <nosv.h>
|
||
|
|
#include <nosv/affinity.h>
|
||
|
|
#include <stdatomic.h>
|
||
|
|
#include <unistd.h>
|
||
|
|
#include "compat.h"
|
||
|
|
#include "common.h"
|
||
|
|
|
||
|
|
#define NTASKS 200
|
||
|
|
#define DEGREE 100
|
||
|
|
|
||
|
|
atomic_int nr_completed_tasks;
|
||
|
|
|
||
|
|
nosv_task_t tasks[NTASKS];
|
||
|
|
nosv_task_t main_task;
|
||
|
|
|
||
|
|
static void
|
||
|
|
task_run(nosv_task_t task)
|
||
|
|
{
|
||
|
|
UNUSED(task);
|
||
|
|
sleep_us(5);
|
||
|
|
}
|
||
|
|
|
||
|
|
static void
|
||
|
|
task_comp(nosv_task_t task)
|
||
|
|
{
|
||
|
|
UNUSED(task);
|
||
|
|
|
||
|
|
int total = atomic_fetch_add(&nr_completed_tasks, 1);
|
||
|
|
if (total == NTASKS - 1) {
|
||
|
|
nosv_submit(main_task, NOSV_SUBMIT_UNLOCKED);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
int
|
||
|
|
main(void)
|
||
|
|
{
|
||
|
|
nosv_init();
|
||
|
|
|
||
|
|
nosv_attach(&main_task, NULL, NULL, NOSV_ATTACH_NONE);
|
||
|
|
|
||
|
|
nosv_task_type_t task_type;
|
||
|
|
nosv_type_init(&task_type, task_run, NULL, task_comp, "task", NULL, NULL, NOSV_TYPE_INIT_NONE);
|
||
|
|
|
||
|
|
for (int t = 0; t < NTASKS; t++) {
|
||
|
|
nosv_create(&tasks[t], task_type, 0, NOSV_CREATE_PARALLEL);
|
||
|
|
nosv_set_task_degree(tasks[t], DEGREE);
|
||
|
|
nosv_submit(tasks[t], NOSV_SUBMIT_NONE);
|
||
|
|
}
|
||
|
|
|
||
|
|
nosv_pause(NOSV_PAUSE_NONE);
|
||
|
|
|
||
|
|
for (int t = 0; t < NTASKS; t++)
|
||
|
|
nosv_destroy(tasks[t], NOSV_DESTROY_NONE);
|
||
|
|
|
||
|
|
nosv_detach(NOSV_DETACH_NONE);
|
||
|
|
|
||
|
|
nosv_type_destroy(task_type, NOSV_TYPE_DESTROY_NONE);
|
||
|
|
|
||
|
|
nosv_shutdown();
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|