Add nOS-V RT parallel task tests
The runtime tests check that we can submit and inline task (which must emit a pause event before the nested task begins) and that we can run parallel tasks.
This commit is contained in:
parent
f81c5695f1
commit
3e5b949c4e
@ -40,6 +40,8 @@ nosv_test(attach.c SORT)
|
||||
nosv_test(waitfor.c SORT)
|
||||
nosv_test(several-tasks.c SORT)
|
||||
nosv_test(init-nested.c SORT)
|
||||
nosv_test(parallel-tasks.c SORT)
|
||||
nosv_test(inline.c SORT)
|
||||
|
||||
# Test multiple instrumentation levels
|
||||
nosv_test(several-tasks.c SORT NAME several-tasks-level-0 LEVEL 0)
|
||||
|
46
test/rt/nosv/inline.c
Normal file
46
test/rt/nosv/inline.c
Normal file
@ -0,0 +1,46 @@
|
||||
/* Copyright (c) 2021-2023 Barcelona Supercomputing Center (BSC)
|
||||
* SPDX-License-Identifier: GPL-3.0-or-later */
|
||||
|
||||
#include <nosv.h>
|
||||
#include "compat.h"
|
||||
#include "common.h"
|
||||
|
||||
static void
|
||||
worker_run(nosv_task_t task)
|
||||
{
|
||||
UNUSED(task);
|
||||
sleep_us(50);
|
||||
}
|
||||
|
||||
static void
|
||||
worker_comp(nosv_task_t task)
|
||||
{
|
||||
UNUSED(task);
|
||||
nosv_destroy(task, NOSV_DESTROY_NONE);
|
||||
}
|
||||
|
||||
/* Runs an inline task inside another one */
|
||||
|
||||
int
|
||||
main(void)
|
||||
{
|
||||
nosv_init();
|
||||
|
||||
nosv_task_t ext_task;
|
||||
if (nosv_attach(&ext_task, NULL, NULL, NOSV_ATTACH_NONE) != 0)
|
||||
die("nosv_attach failed");
|
||||
|
||||
nosv_task_type_t worker;
|
||||
if (nosv_type_init(&worker, worker_run, NULL, worker_comp, "worker", NULL, NULL, NOSV_TYPE_INIT_NONE) != 0)
|
||||
die("nosv_type_init failed");
|
||||
|
||||
nosv_task_t worker_task;
|
||||
nosv_create(&worker_task, worker, 0, NOSV_CREATE_NONE);
|
||||
|
||||
nosv_submit(worker_task, NOSV_SUBMIT_INLINE);
|
||||
nosv_type_destroy(worker, NOSV_TYPE_DESTROY_NONE);
|
||||
|
||||
nosv_detach(NOSV_DETACH_NONE);
|
||||
|
||||
nosv_shutdown();
|
||||
}
|
65
test/rt/nosv/parallel-tasks.c
Normal file
65
test/rt/nosv/parallel-tasks.c
Normal file
@ -0,0 +1,65 @@
|
||||
/* 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;
|
||||
}
|
Loading…
Reference in New Issue
Block a user