Files
mlib/tests/test-mworker.c
T

155 lines
3.9 KiB
C

/*
* Copyright (c) 2017-2025, Patrick Pelissier
* All rights reserved.
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
* + Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* + Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND ANY
* EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL THE REGENTS AND CONTRIBUTORS BE LIABLE FOR ANY
* DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
#include "m-core.h"
#include "coverage.h"
#include "m-worker.h"
/* Compute Fibonacci number using thread systems. */
static worker_t w_g;
static int fib(int n);
struct fib2_s {
int x, n;
};
atomic_bool resetFunc_called = M_ATOMIC_VAR_INIT(false);
static void resetFunc(void)
{
atomic_store(&resetFunc_called, true);
}
static void subfunc_1 (void *data) {
struct fib2_s *f = M_ASSIGN_CAST (struct fib2_s *, data);
f->x = fib (f->n );
}
static int fib(int n)
{
if (n < 2)
return n;
struct fib2_s f;
worker_sync_t b;
worker_start(b, w_g);
f.n = n - 2;
worker_spawn (b, subfunc_1, &f);
int y = fib (n-1);
worker_sync(b);
return f.x + y;
}
static void test1(void)
{
atomic_store(&resetFunc_called, false);
worker_init(w_g, 0, 0, resetFunc);
int result = fib(39);
assert (result == 63245986);
worker_clear(w_g);
assert (atomic_load(&resetFunc_called) == true || m_work3r_get_cpu_count() == 1);
}
static void test1bis(void)
{
atomic_store(&resetFunc_called, false);
worker_init(w_g, 0, 0, NULL, resetFunc);
int result = fib(4);
assert (result == 3);
worker_clear(w_g);
assert (atomic_load(&resetFunc_called) == true || m_work3r_get_cpu_count() == 1);
}
#if defined(__GNUC__) && (!defined(__clang__) || (defined(WORKER_USE_CLANG_BLOCK) && WORKER_USE_CLANG_BLOCK) || (defined(WORKER_USE_CPP_FUNCTION) && WORKER_USE_CPP_FUNCTION))
/* The macro version will generate warnings about shadow variables.
There is no way to avoid this. */
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wshadow"
static int fib2(int n)
{
if (n < 2)
return n;
worker_sync_t b;
int y1, y2;
worker_start(b, w_g);
WORKER_SPAWN(b, (n), { y2 = fib2(n-2); }, (y2));
y1 = fib (n-1);
worker_sync(b);
return y1 + y2;
}
#pragma GCC diagnostic pop
static void test2(void)
{
worker_init(w_g, 0, 0, NULL);
int result = fib2(39);
assert (result == 63245986);
worker_clear(w_g);
}
#else
static void test2(void) {}
#endif
// Test using specialization of worker_spawn
M_WORKER_SPAWN_DEF2(fib, (out, int *, M_PTR_OPLIST), (in, int) )
static int fib3(int);
static void subfib3 (int *dst, int n) {
*dst = fib3 (n);
}
static int fib3(int n)
{
if (n < 2)
return n;
worker_sync_t b;
worker_start(b, w_g);
int x;
m_worker_spawn_fib (b, subfib3, &x, n-2);
int y = fib (n-1);
worker_sync(b);
return x + y;
}
static void test3(void)
{
worker_init(w_g, 0);
int result = fib3(39);
assert (result == 63245986);
worker_clear(w_g);
}
int main(void)
{
test1();
test1bis();
test2();
test3();
exit(0);
}