ovni/test/rt/nosv/yield-noisy.c
Rodrigo Arias c847ec19ad Add noisy yield test in nOS-V
Ensure we can call nosv_yield in a loop without generating a lot of
events when using the level 3 instrumentation level. Noisy events are
now moved to level 4, so they shouldn't appear on the trace.
Additionally, make sure that the noisy events appear on level 4.
2025-03-21 16:57:44 +01:00

90 lines
1.5 KiB
C

/* Copyright (c) 2025 Barcelona Supercomputing Center (BSC)
* SPDX-License-Identifier: GPL-3.0-or-later */
#include <nosv.h>
#include <nosv/alpi.h>
#include <stdatomic.h>
#include <unistd.h>
#include <stdlib.h>
#include "common.h"
#include "compat.h"
#define NITERS 1000L
atomic_int ncompleted = 0;
static void
busywait(long iter)
{
for (volatile long i = 0; i < iter; i++)
;
}
static void
task_body(nosv_task_t task)
{
UNUSED(task);
/* Yield a lot of times to try to generate several events */
for (long i = 0; i < NITERS; i++) {
nosv_yield(0);
busywait(10000L);
}
}
static void
task_done(nosv_task_t task)
{
UNUSED(task);
atomic_fetch_add(&ncompleted, 1);
}
int
main(void)
{
nosv_init();
uint64_t ncpus;
if (alpi_cpu_count(&ncpus))
die("alpi_cpu_count failed");
nosv_task_t *tasks = calloc((size_t) ncpus, sizeof(nosv_task_t));
if (tasks == NULL)
die("calloc failed:");
int ntasks = (int) ncpus;
info("ntasks = %d", ntasks);
FILE *f = fopen("vars.sh", "w");
if (f == NULL)
die("fopen failed:");
fprintf(f, "nyields=%ld\n", (long) ntasks * NITERS);
fclose(f);
nosv_task_type_t task_type;
nosv_type_init(&task_type, task_body, NULL, task_done, "task", NULL, NULL, 0);
for (int i = 0; i < ntasks; i++)
nosv_create(&tasks[i], task_type, 0, 0);
for (int i = 0; i < ntasks; i++)
nosv_submit(tasks[i], 0);
while (atomic_load(&ncompleted) != ntasks)
sleep_us(1000);
for (int i = 0; i < ntasks; i++)
nosv_destroy(tasks[i], 0);
nosv_type_destroy(task_type, 0);
free(tasks);
nosv_shutdown();
return 0;
}