/* * Unit test suite for vcomp * * Copyright 2012 Dan Kegel * Copyright 2015-2016 Sebastian Lackner * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA */ #include #include "wine/test.h" static char vcomp_manifest_file[MAX_PATH]; static HANDLE vcomp_actctx_hctx; static ULONG_PTR vcomp_actctx_cookie; static HMODULE vcomp_handle; static HANDLE (WINAPI *pCreateActCtxA)(ACTCTXA*); static BOOL (WINAPI *pActivateActCtx)(HANDLE, ULONG_PTR*); static BOOL (WINAPI *pDeactivateActCtx)(DWORD, ULONG_PTR); static VOID (WINAPI *pReleaseActCtx)(HANDLE); typedef CRITICAL_SECTION *omp_lock_t; typedef CRITICAL_SECTION *omp_nest_lock_t; static void (CDECL *p_vcomp_atomic_add_i4)(int *dest, int val); static void (CDECL *p_vcomp_atomic_add_i8)(LONG64 *dest, LONG64 val); static void (CDECL *p_vcomp_atomic_add_r4)(float *dest, float val); static void (CDECL *p_vcomp_atomic_add_r8)(double *dest, double val); static void (CDECL *p_vcomp_atomic_and_i4)(int *dest, int val); static void (CDECL *p_vcomp_atomic_and_i8)(LONG64 *dest, LONG64 val); static void (CDECL *p_vcomp_atomic_div_i4)(int *dest, int val); static void (CDECL *p_vcomp_atomic_div_i8)(LONG64 *dest, LONG64 val); static void (CDECL *p_vcomp_atomic_div_r4)(float *dest, float val); static void (CDECL *p_vcomp_atomic_div_r8)(double *dest, double val); static void (CDECL *p_vcomp_atomic_div_ui4)(unsigned int *dest, unsigned int val); static void (CDECL *p_vcomp_atomic_div_ui8)(ULONG64 *dest, ULONG64 val); static void (CDECL *p_vcomp_atomic_mul_i4)(int *dest, int val); static void (CDECL *p_vcomp_atomic_mul_i8)(LONG64 *dest, LONG64 val); static void (CDECL *p_vcomp_atomic_mul_r4)(float *dest, float val); static void (CDECL *p_vcomp_atomic_mul_r8)(double *dest, double val); static void (CDECL *p_vcomp_atomic_or_i4)(int *dest, int val); static void (CDECL *p_vcomp_atomic_or_i8)(LONG64 *dest, LONG64 val); static void (CDECL *p_vcomp_atomic_shl_i4)(int *dest, int val); static void (CDECL *p_vcomp_atomic_shl_i8)(LONG64 *dest, unsigned int val); static void (CDECL *p_vcomp_atomic_shr_i4)(int *dest, int val); static void (CDECL *p_vcomp_atomic_shr_i8)(LONG64 *dest, unsigned int val); static void (CDECL *p_vcomp_atomic_shr_ui4)(unsigned int *dest, unsigned int val); static void (CDECL *p_vcomp_atomic_shr_ui8)(ULONG64 *dest, unsigned int val); static void (CDECL *p_vcomp_atomic_sub_i4)(int *dest, int val); static void (CDECL *p_vcomp_atomic_sub_i8)(LONG64 *dest, LONG64 val); static void (CDECL *p_vcomp_atomic_sub_r4)(float *dest, float val); static void (CDECL *p_vcomp_atomic_sub_r8)(double *dest, double val); static void (CDECL *p_vcomp_atomic_xor_i4)(int *dest, int val); static void (CDECL *p_vcomp_atomic_xor_i8)(LONG64 *dest, LONG64 val); static void (CDECL *p_vcomp_barrier)(void); static void (CDECL *p_vcomp_enter_critsect)(CRITICAL_SECTION **critsect); static void (CDECL *p_vcomp_flush)(void); static void (CDECL *p_vcomp_for_dynamic_init)(unsigned int flags, unsigned int first, unsigned int last, int step, unsigned int chunksize); static int (CDECL *p_vcomp_for_dynamic_next)(unsigned int *begin, unsigned int *end); static void (CDECL *p_vcomp_for_static_end)(void); static void (CDECL *p_vcomp_for_static_init)(int first, int last, int step, int chunksize, unsigned int *loops, int *begin, int *end, int *next, int *lastchunk); static void (CDECL *p_vcomp_for_static_simple_init)(unsigned int first, unsigned int last, int step, BOOL increment, unsigned int *begin, unsigned int *end); static void (WINAPIV *p_vcomp_fork)(BOOL ifval, int nargs, void *wrapper, ...); static int (CDECL *p_vcomp_get_thread_num)(void); static void (CDECL *p_vcomp_leave_critsect)(CRITICAL_SECTION *critsect); static int (CDECL *p_vcomp_master_begin)(void); static void (CDECL *p_vcomp_master_end)(void); static void (CDECL *p_vcomp_sections_init)(int n); static int (CDECL *p_vcomp_sections_next)(void); static void (CDECL *p_vcomp_set_num_threads)(int num_threads); static int (CDECL *p_vcomp_single_begin)(int flags); static void (CDECL *p_vcomp_single_end)(void); static void (CDECL *pomp_destroy_lock)(omp_lock_t *lock); static void (CDECL *pomp_destroy_nest_lock)(omp_nest_lock_t *lock); static int (CDECL *pomp_get_max_threads)(void); static int (CDECL *pomp_get_nested)(void); static int (CDECL *pomp_get_num_threads)(void); static int (CDECL *pomp_get_thread_num)(void); static int (CDECL *pomp_in_parallel)(void); static void (CDECL *pomp_init_lock)(omp_lock_t *lock); static void (CDECL *pomp_init_nest_lock)(omp_nest_lock_t *lock); static void (CDECL *pomp_set_lock)(omp_lock_t *lock); static void (CDECL *pomp_set_nest_lock)(omp_nest_lock_t *lock); static void (CDECL *pomp_set_nested)(int nested); static void (CDECL *pomp_set_num_threads)(int num_threads); static int (CDECL *pomp_test_lock)(omp_lock_t *lock); static int (CDECL *pomp_test_nest_lock)(omp_nest_lock_t *lock); static void (CDECL *pomp_unset_lock)(omp_lock_t *lock); static void (CDECL *pomp_unset_nest_lock)(omp_nest_lock_t *lock); #define VCOMP_DYNAMIC_FLAGS_STATIC 0x01 #define VCOMP_DYNAMIC_FLAGS_CHUNKED 0x02 #define VCOMP_DYNAMIC_FLAGS_GUIDED 0x03 #define VCOMP_DYNAMIC_FLAGS_INCREMENT 0x40 #ifdef __i386__ #define ARCH "x86" #elif defined(__x86_64__) #define ARCH "amd64" #elif defined __arm__ #define ARCH "arm" #elif defined __aarch64__ #define ARCH "arm64" #else #define ARCH "none" #endif static const char vcomp_manifest[] = "\n" "\n" " \n" "Wine vcomp test suite\n" "\n" " \n" " \n" " \n" "\n" "\n"; #undef ARCH static const char *debugstr_longlong(ULONGLONG ll) { static char str[17]; if (sizeof(ll) > sizeof(unsigned long) && ll >> 32) sprintf(str, "%lx%08lx", (unsigned long)(ll >> 32), (unsigned long)ll); else sprintf(str, "%lx", (unsigned long)ll); return str; } static void create_vcomp_manifest(void) { char temp_path[MAX_PATH]; HMODULE kernel32; DWORD written; ACTCTXA ctx; HANDLE file; kernel32 = GetModuleHandleA("kernel32.dll"); pCreateActCtxA = (void *)GetProcAddress(kernel32, "CreateActCtxA"); pActivateActCtx = (void *)GetProcAddress(kernel32, "ActivateActCtx"); pDeactivateActCtx = (void *)GetProcAddress(kernel32, "DeactivateActCtx"); pReleaseActCtx = (void *)GetProcAddress(kernel32, "ReleaseActCtx"); if (!pCreateActCtxA) return; if (!GetTempPathA(sizeof(temp_path), temp_path) || !GetTempFileNameA(temp_path, "vcomp", 0, vcomp_manifest_file)) { ok(0, "failed to create manifest file\n"); return; } file = CreateFileA(vcomp_manifest_file, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, NULL); if (file == INVALID_HANDLE_VALUE) { ok(0, "failed to open manifest file\n"); return; } if (!WriteFile(file, vcomp_manifest, sizeof(vcomp_manifest) - 1, &written, NULL)) written = 0; CloseHandle(file); if (written != sizeof(vcomp_manifest) - 1) { ok(0, "failed to write manifest file\n"); DeleteFileA(vcomp_manifest_file); return; } memset(&ctx, 0, sizeof(ctx)); ctx.cbSize = sizeof(ctx); ctx.lpSource = vcomp_manifest_file; vcomp_actctx_hctx = pCreateActCtxA(&ctx); if (!vcomp_actctx_hctx) { ok(0, "failed to create activation context\n"); DeleteFileA(vcomp_manifest_file); return; } if (!pActivateActCtx(vcomp_actctx_hctx, &vcomp_actctx_cookie)) { win_skip("failed to activate context\n"); pReleaseActCtx(vcomp_actctx_hctx); DeleteFileA(vcomp_manifest_file); vcomp_actctx_hctx = NULL; } } static void release_vcomp(void) { if (vcomp_handle) FreeLibrary(vcomp_handle); if (vcomp_actctx_hctx) { pDeactivateActCtx(0, vcomp_actctx_cookie); pReleaseActCtx(vcomp_actctx_hctx); DeleteFileA(vcomp_manifest_file); } } #define VCOMP_GET_PROC(func) \ do \ { \ p ## func = (void *)GetProcAddress(vcomp_handle, #func); \ if (!p ## func) trace("Failed to get address for %s\n", #func); \ } \ while (0) static BOOL init_vcomp(void) { create_vcomp_manifest(); vcomp_handle = LoadLibraryA("vcomp.dll"); if (!vcomp_handle) { win_skip("vcomp.dll not installed\n"); release_vcomp(); return FALSE; } VCOMP_GET_PROC(_vcomp_atomic_add_i4); VCOMP_GET_PROC(_vcomp_atomic_add_i8); VCOMP_GET_PROC(_vcomp_atomic_add_r4); VCOMP_GET_PROC(_vcomp_atomic_add_r8); VCOMP_GET_PROC(_vcomp_atomic_and_i4); VCOMP_GET_PROC(_vcomp_atomic_and_i8); VCOMP_GET_PROC(_vcomp_atomic_div_i4); VCOMP_GET_PROC(_vcomp_atomic_div_i8); VCOMP_GET_PROC(_vcomp_atomic_div_r4); VCOMP_GET_PROC(_vcomp_atomic_div_r8); VCOMP_GET_PROC(_vcomp_atomic_div_ui4); VCOMP_GET_PROC(_vcomp_atomic_div_ui8); VCOMP_GET_PROC(_vcomp_atomic_mul_i4); VCOMP_GET_PROC(_vcomp_atomic_mul_i8); VCOMP_GET_PROC(_vcomp_atomic_mul_r4); VCOMP_GET_PROC(_vcomp_atomic_mul_r8); VCOMP_GET_PROC(_vcomp_atomic_or_i4); VCOMP_GET_PROC(_vcomp_atomic_or_i8); VCOMP_GET_PROC(_vcomp_atomic_shl_i4); VCOMP_GET_PROC(_vcomp_atomic_shl_i8); VCOMP_GET_PROC(_vcomp_atomic_shr_i4); VCOMP_GET_PROC(_vcomp_atomic_shr_i8); VCOMP_GET_PROC(_vcomp_atomic_shr_ui4); VCOMP_GET_PROC(_vcomp_atomic_shr_ui8); VCOMP_GET_PROC(_vcomp_atomic_sub_i4); VCOMP_GET_PROC(_vcomp_atomic_sub_i8); VCOMP_GET_PROC(_vcomp_atomic_sub_r4); VCOMP_GET_PROC(_vcomp_atomic_sub_r8); VCOMP_GET_PROC(_vcomp_atomic_xor_i4); VCOMP_GET_PROC(_vcomp_atomic_xor_i8); VCOMP_GET_PROC(_vcomp_barrier); VCOMP_GET_PROC(_vcomp_enter_critsect); VCOMP_GET_PROC(_vcomp_flush); VCOMP_GET_PROC(_vcomp_for_dynamic_init); VCOMP_GET_PROC(_vcomp_for_dynamic_next); VCOMP_GET_PROC(_vcomp_for_static_end); VCOMP_GET_PROC(_vcomp_for_static_init); VCOMP_GET_PROC(_vcomp_for_static_simple_init); VCOMP_GET_PROC(_vcomp_fork); VCOMP_GET_PROC(_vcomp_get_thread_num); VCOMP_GET_PROC(_vcomp_leave_critsect); VCOMP_GET_PROC(_vcomp_master_begin); VCOMP_GET_PROC(_vcomp_master_end); VCOMP_GET_PROC(_vcomp_sections_init); VCOMP_GET_PROC(_vcomp_sections_next); VCOMP_GET_PROC(_vcomp_set_num_threads); VCOMP_GET_PROC(_vcomp_single_begin); VCOMP_GET_PROC(_vcomp_single_end); VCOMP_GET_PROC(omp_destroy_lock); VCOMP_GET_PROC(omp_destroy_nest_lock); VCOMP_GET_PROC(omp_get_max_threads); VCOMP_GET_PROC(omp_get_nested); VCOMP_GET_PROC(omp_get_num_threads); VCOMP_GET_PROC(omp_get_thread_num); VCOMP_GET_PROC(omp_in_parallel); VCOMP_GET_PROC(omp_init_lock); VCOMP_GET_PROC(omp_init_nest_lock); VCOMP_GET_PROC(omp_set_lock); VCOMP_GET_PROC(omp_set_nest_lock); VCOMP_GET_PROC(omp_set_nested); VCOMP_GET_PROC(omp_set_num_threads); VCOMP_GET_PROC(omp_test_lock); VCOMP_GET_PROC(omp_test_nest_lock); VCOMP_GET_PROC(omp_unset_lock); VCOMP_GET_PROC(omp_unset_nest_lock); return TRUE; } #undef VCOMP_GET_PROC static void CDECL num_threads_cb2(int parallel, LONG *count) { int is_parallel = pomp_in_parallel(); ok(is_parallel == parallel, "expected %d, got %d\n", parallel, is_parallel); InterlockedIncrement(count); } static void CDECL num_threads_cb(BOOL nested, int parallel, int nested_threads, LONG *count) { int is_parallel, num_threads, thread_num; LONG thread_count; InterlockedIncrement(count); p_vcomp_barrier(); num_threads = pomp_get_num_threads(); ok(num_threads == *count, "expected num_threads == %d, got %d\n", *count, num_threads); thread_num = pomp_get_thread_num(); ok(thread_num >= 0 && thread_num < num_threads, "expected thread_num in range [0, %d], got %d\n", num_threads - 1, thread_num); ok(thread_num == p_vcomp_get_thread_num(), "expected _vcomp_get_thread_num to return the same value\n"); is_parallel = pomp_in_parallel(); ok(is_parallel == parallel, "expected %d, got %d\n", parallel, is_parallel); thread_count = 0; p_vcomp_fork(TRUE, 2, num_threads_cb2, TRUE, &thread_count); if (nested) ok(thread_count == nested_threads, "expected %d threads, got %d\n", nested_threads, thread_count); else ok(thread_count == 1, "expected 1 thread, got %d\n", thread_count); is_parallel = pomp_in_parallel(); ok(is_parallel == parallel, "expected %d, got %d\n", parallel, is_parallel); thread_count = 0; p_vcomp_fork(FALSE, 2, num_threads_cb2, parallel, &thread_count); ok(thread_count == 1, "expected 1 thread, got %d\n", thread_count); is_parallel = pomp_in_parallel(); ok(is_parallel == parallel, "expected %d, got %d\n", parallel, is_parallel); p_vcomp_set_num_threads(4); thread_count = 0; p_vcomp_fork(TRUE, 2, num_threads_cb2, TRUE, &thread_count); if (nested) ok(thread_count == 4, "expected 4 threads, got %d\n", thread_count); else ok(thread_count == 1, "expected 1 thread, got %d\n", thread_count); is_parallel = pomp_in_parallel(); ok(is_parallel == parallel, "expected %d, got %d\n", parallel, is_parallel); } static void test_omp_get_num_threads(BOOL nested) { int is_nested, is_parallel, max_threads, num_threads, thread_num; LONG thread_count; ok(pomp_get_thread_num != p_vcomp_get_thread_num, "expected omp_get_thread_num != _vcomp_get_thread_num\n"); pomp_set_nested(nested); is_nested = pomp_get_nested(); ok(is_nested == nested, "expected %d, got %d\n", nested, is_nested); max_threads = pomp_get_max_threads(); ok(max_threads >= 1, "expected max_threads >= 1, got %d\n", max_threads); thread_num = pomp_get_thread_num(); ok(thread_num == 0, "expected thread_num == 0, got %d\n", thread_num); is_parallel = pomp_in_parallel(); ok(is_parallel == FALSE, "expected FALSE, got %d\n", is_parallel); num_threads = pomp_get_num_threads(); ok(num_threads == 1, "expected num_threads == 1, got %d\n", num_threads); thread_count = 0; p_vcomp_fork(TRUE, 4, num_threads_cb, nested, TRUE, max_threads, &thread_count); ok(thread_count == max_threads, "expected %d threads, got %d\n", max_threads, thread_count); is_parallel = pomp_in_parallel(); ok(is_parallel == FALSE, "expected FALSE, got %d\n", is_parallel); num_threads = pomp_get_num_threads(); ok(num_threads == 1, "expected num_threads == 1, got %d\n", num_threads); thread_count = 0; p_vcomp_fork(FALSE, 4, num_threads_cb, TRUE, FALSE, max_threads, &thread_count); ok(thread_count == 1, "expected 1 thread, got %d\n", thread_count); is_parallel = pomp_in_parallel(); ok(is_parallel == FALSE, "expected FALSE, got %d\n", is_parallel); pomp_set_num_threads(1); num_threads = pomp_get_num_threads(); ok(num_threads == 1, "expected num_threads == 1, got %d\n", num_threads); thread_count = 0; p_vcomp_fork(TRUE, 4, num_threads_cb, nested, TRUE, 1, &thread_count); ok(thread_count == 1, "expected 1 thread, got %d\n", thread_count); is_parallel = pomp_in_parallel(); ok(is_parallel == FALSE, "expected FALSE, got %d\n", is_parallel); pomp_set_num_threads(2); num_threads = pomp_get_num_threads(); ok(num_threads == 1, "expected num_threads == 1, got %d\n", num_threads); thread_count = 0; p_vcomp_fork(TRUE, 4, num_threads_cb, nested, TRUE, 2, &thread_count); ok(thread_count == 2, "expected 2 threads, got %d\n", thread_count); pomp_set_num_threads(4); num_threads = pomp_get_num_threads(); ok(num_threads == 1, "expected num_threads == 1, got %d\n", num_threads); thread_count = 0; p_vcomp_fork(TRUE, 4, num_threads_cb, nested, TRUE, 4, &thread_count); ok(thread_count == 4, "expected 4 threads, got %d\n", thread_count); p_vcomp_set_num_threads(8); num_threads = pomp_get_num_threads(); ok(num_threads == 1, "expected num_threads == 1, got %d\n", num_threads); thread_count = 0; p_vcomp_fork(TRUE, 4, num_threads_cb, nested, TRUE, 4, &thread_count); ok(thread_count == 8, "expected 8 threads, got %d\n", thread_count); thread_count = 0; p_vcomp_fork(TRUE, 4, num_threads_cb, nested, TRUE, 4, &thread_count); ok(thread_count == 4, "expected 4 threads, got %d\n", thread_count); p_vcomp_set_num_threads(0); num_threads = pomp_get_num_threads(); ok(num_threads == 1, "expected num_threads == 1, got %d\n", num_threads); thread_count = 0; p_vcomp_fork(TRUE, 4, num_threads_cb, nested, TRUE, 4, &thread_count); ok(thread_count == 4, "expected 4 threads, got %d\n", thread_count); pomp_set_num_threads(0); num_threads = pomp_get_num_threads(); ok(num_threads == 1, "expected num_threads == 1, got %d\n", num_threads); thread_count = 0; p_vcomp_fork(TRUE, 4, num_threads_cb, nested, TRUE, 4, &thread_count); ok(thread_count == 4, "expected 4 threads, got %d\n", thread_count); pomp_set_num_threads(max_threads); pomp_set_nested(FALSE); } static void CDECL fork_ptr_cb(LONG *a, LONG *b, LONG *c, LONG *d, LONG *e) { InterlockedIncrement(a); InterlockedIncrement(b); InterlockedIncrement(c); InterlockedIncrement(d); InterlockedIncrement(e); } static void CDECL fork_uintptr_cb(UINT_PTR a, UINT_PTR b, UINT_PTR c, UINT_PTR d, UINT_PTR e) { ok(a == 1, "expected a == 1, got %p\n", (void *)a); ok(b == MAXUINT_PTR - 2, "expected b == MAXUINT_PTR - 2, got %p\n", (void *)b); ok(c == 3, "expected c == 3, got %p\n", (void *)c); ok(d == MAXUINT_PTR - 4, "expected d == MAXUINT_PTR - 4, got %p\n", (void *)d); ok(e == 5, "expected e == 5, got %p\n", (void *)e); } #ifdef __i386__ static void CDECL fork_float_cb(float a, float b, float c, float d, float e) { ok(1.4999 < a && a < 1.5001, "expected a == 1.5, got %f\n", a); ok(2.4999 < b && b < 2.5001, "expected b == 2.5, got %f\n", b); ok(3.4999 < c && c < 3.5001, "expected c == 3.5, got %f\n", c); ok(4.4999 < d && d < 4.5001, "expected d == 4.5, got %f\n", d); ok(5.4999 < e && e < 5.5001, "expected e == 5.5, got %f\n", e); } #endif static void test_vcomp_fork(void) { LONG a, b, c, d, e; int max_threads = pomp_get_max_threads(); pomp_set_num_threads(4); a = 0; b = 1; c = 2; d = 3; e = 4; p_vcomp_fork(FALSE, 5, fork_ptr_cb, &a, &b, &c, &d, &e); ok(a == 1, "expected a == 1, got %d\n", a); ok(b == 2, "expected b == 2, got %d\n", b); ok(c == 3, "expected c == 3, got %d\n", c); ok(d == 4, "expected d == 4, got %d\n", d); ok(e == 5, "expected e == 5, got %d\n", e); a = 0; b = 1; c = 2; d = 3; e = 4; p_vcomp_fork(TRUE, 5, fork_ptr_cb, &a, &b, &c, &d, &e); ok(a == 4, "expected a == 4, got %d\n", a); ok(b == 5, "expected b == 5, got %d\n", b); ok(c == 6, "expected c == 6, got %d\n", c); ok(d == 7, "expected d == 7, got %d\n", d); ok(e == 8, "expected e == 8, got %d\n", e); p_vcomp_fork(TRUE, 5, fork_uintptr_cb, (UINT_PTR)1, (UINT_PTR)(MAXUINT_PTR - 2), (UINT_PTR)3, (UINT_PTR)(MAXUINT_PTR - 4), (UINT_PTR)5); #ifdef __i386__ { void (CDECL *func)(BOOL, int, void *, float, float, float, float, float) = (void *)p_vcomp_fork; func(TRUE, 5, fork_float_cb, 1.5f, 2.5f, 3.5f, 4.5f, 5.5f); } #else skip("skipping float test on non-x86\n"); #endif pomp_set_num_threads(max_threads); } static void CDECL section_cb(LONG *a, LONG *b, LONG *c) { int i; p_vcomp_sections_init(20); while ((i = p_vcomp_sections_next()) != -1) { InterlockedIncrement(a); Sleep(1); } p_vcomp_sections_init(30); while ((i = p_vcomp_sections_next()) != -1) { InterlockedIncrement(b); Sleep(1); } p_vcomp_sections_init(40); while ((i = p_vcomp_sections_next()) != -1) { InterlockedIncrement(c); Sleep(1); } } static void test_vcomp_sections_init(void) { LONG a, b, c; int max_threads = pomp_get_max_threads(); int i; if (0) { /* calling _vcomp_sections_next without prior _vcomp_sections_init * returns uninitialized memory on Windows. */ i = p_vcomp_sections_next(); ok(i == -1, "expected -1, got %d\n", i); } a = b = c = 0; section_cb(&a, &b, &c); ok(a == 20, "expected a == 20, got %d\n", a); ok(b == 30, "expected b == 30, got %d\n", b); ok(c == 40, "expected c == 40, got %d\n", c); for (i = 1; i <= 4; i++) { pomp_set_num_threads(i); a = b = c = 0; p_vcomp_fork(TRUE, 3, section_cb, &a, &b, &c); ok(a == 20, "expected a == 20, got %d\n", a); ok(b == 30, "expected b == 30, got %d\n", b); ok(c == 40, "expected c == 40, got %d\n", c); a = b = c = 0; p_vcomp_fork(FALSE, 3, section_cb, &a, &b, &c); ok(a == 20, "expected a == 20, got %d\n", a); ok(b == 30, "expected b == 30, got %d\n", b); ok(c == 40, "expected c == 40, got %d\n", c); } pomp_set_num_threads(max_threads); } static void my_for_static_simple_init(BOOL dynamic, unsigned int first, unsigned int last, int step, BOOL increment, unsigned int *begin, unsigned int *end) { unsigned int iterations, per_thread, remaining; int num_threads = pomp_get_num_threads(); int thread_num = pomp_get_thread_num(); if (!dynamic && num_threads == 1) { *begin = first; *end = last; return; } if (step <= 0) { *begin = 0; *end = increment ? -1 : 1; return; } if (increment) iterations = 1 + (last - first) / step; else { iterations = 1 + (first - last) / step; step *= -1; } per_thread = iterations / num_threads; remaining = iterations - per_thread * num_threads; if (thread_num < remaining) per_thread++; else if (per_thread) first += remaining * step; else { *begin = first; *end = first - step; return; } *begin = first + per_thread * thread_num * step; *end = *begin + (per_thread - 1) * step; } static void CDECL for_static_simple_cb(void) { static const struct { unsigned int first; unsigned int last; int step; } tests[] = { { 0, 0, 1 }, /* 0 */ { 0, 1, 1 }, { 0, 2, 1 }, { 0, 3, 1 }, { 0, 100, 0 }, { 0, 100, 1 }, { 0, 100, 2 }, { 0, 100, 3 }, { 0, 100, -1 }, { 0, 100, -2 }, { 0, 100, -3 }, /* 10 */ { 0, 100, 10 }, { 0, 100, 50 }, { 0, 100, 100 }, { 0, 100, 150 }, { 0, 0x80000000, 1 }, { 0, 0xfffffffe, 1 }, { 0, 0xffffffff, 1 }, { 50, 50, 0 }, { 50, 50, 1 }, { 50, 50, 2 }, /* 20 */ { 50, 50, 3 }, { 50, 50, -1 }, { 50, 50, -2 }, { 50, 50, -3 }, { 100, 200, 1 }, { 100, 200, 5 }, { 100, 200, 10 }, { 100, 200, 50 }, { 100, 200, 100 }, { 100, 200, 150 }, /* 30 */ }; int num_threads = pomp_get_num_threads(); int thread_num = pomp_get_thread_num(); int i; for (i = 0; i < sizeof(tests)/sizeof(tests[0]); i++) { unsigned int my_begin, my_end, begin, end; begin = end = 0xdeadbeef; my_for_static_simple_init(FALSE, tests[i].first, tests[i].last, tests[i].step, FALSE, &my_begin, &my_end); p_vcomp_for_static_simple_init(tests[i].first, tests[i].last, tests[i].step, FALSE, &begin, &end); ok(begin == my_begin, "test %d, thread %d/%d: expected begin == %u, got %u\n", i, thread_num, num_threads, my_begin, begin); ok(end == my_end, "test %d, thread %d/%d: expected end == %u, got %u\n", i, thread_num, num_threads, my_end, end); p_vcomp_for_static_end(); p_vcomp_barrier(); begin = end = 0xdeadbeef; my_for_static_simple_init(FALSE, tests[i].first, tests[i].last, tests[i].step, TRUE, &my_begin, &my_end); p_vcomp_for_static_simple_init(tests[i].first, tests[i].last, tests[i].step, TRUE, &begin, &end); ok(begin == my_begin, "test %d, thread %d/%d: expected begin == %u, got %u\n", i, thread_num, num_threads, my_begin, begin); ok(end == my_end, "test %d, thread %d/%d: expected end == %u, got %u\n", i, thread_num, num_threads, my_end, end); p_vcomp_for_static_end(); p_vcomp_barrier(); if (tests[i].first == tests[i].last) continue; begin = end = 0xdeadbeef; my_for_static_simple_init(FALSE, tests[i].last, tests[i].first, tests[i].step, FALSE, &my_begin, &my_end); p_vcomp_for_static_simple_init(tests[i].last, tests[i].first, tests[i].step, FALSE, &begin, &end); ok(begin == my_begin, "test %d, thread %d/%d: expected begin == %u, got %u\n", i, thread_num, num_threads, my_begin, begin); ok(end == my_end, "test %d, thread %d/%d: expected end == %u, got %u\n", i, thread_num, num_threads, my_end, end); p_vcomp_for_static_end(); p_vcomp_barrier(); begin = end = 0xdeadbeef; my_for_static_simple_init(FALSE, tests[i].last, tests[i].first, tests[i].step, TRUE, &my_begin, &my_end); p_vcomp_for_static_simple_init(tests[i].last, tests[i].first, tests[i].step, TRUE, &begin, &end); ok(begin == my_begin, "test %d, thread %d/%d: expected begin == %u, got %u\n", i, thread_num, num_threads, my_begin, begin); ok(end == my_end, "test %d, thread %d/%d: expected end == %u, got %u\n", i, thread_num, num_threads, my_end, end); p_vcomp_for_static_end(); p_vcomp_barrier(); } } static void test_vcomp_for_static_simple_init(void) { int max_threads = pomp_get_max_threads(); int i; for_static_simple_cb(); for (i = 1; i <= 4; i++) { pomp_set_num_threads(i); p_vcomp_fork(TRUE, 0, for_static_simple_cb); p_vcomp_fork(FALSE, 0, for_static_simple_cb); } pomp_set_num_threads(max_threads); } #define VCOMP_FOR_STATIC_BROKEN_LOOP 1 #define VCOMP_FOR_STATIC_BROKEN_NEXT 2 static DWORD CDECL my_for_static_init(int first, int last, int step, int chunksize, unsigned int *loops, int *begin, int *end, int *next, int *lastchunk) { unsigned int iterations, num_chunks, per_thread, remaining; int num_threads = pomp_get_num_threads(); int thread_num = pomp_get_thread_num(); if (num_threads == 1 && chunksize != 1) { *loops = 1; *begin = first; *end = last; *next = 0; *lastchunk = first; return 0; } if (first == last) { *loops = !thread_num; if (!thread_num) { /* The value in *next on Windows is either uninitialized, or contains * garbage. The value shouldn't matter for *loops <= 1, so no need to * reproduce that. */ *begin = first; *end = last; *next = 0; *lastchunk = first; } return thread_num ? 0 : VCOMP_FOR_STATIC_BROKEN_NEXT; } if (step <= 0) { /* The total number of iterations depends on the number of threads here, * which doesn't make any sense. This is most likely a bug in the Windows * implementation. */ return VCOMP_FOR_STATIC_BROKEN_LOOP; } if (first < last) iterations = 1 + (last - first) / step; else { iterations = 1 + (first - last) / step; step *= -1; } if (chunksize < 1) chunksize = 1; num_chunks = ((DWORD64)iterations + chunksize - 1) / chunksize; per_thread = num_chunks / num_threads; remaining = num_chunks - per_thread * num_threads; *loops = per_thread + (thread_num < remaining); *begin = first + thread_num * chunksize * step; *end = *begin + (chunksize - 1) * step; *next = chunksize * num_threads * step; *lastchunk = first + (num_chunks - 1) * chunksize * step; return 0; } static void CDECL for_static_cb(void) { static const struct { int first; int last; int step; int chunksize; } tests[] = { { 0, 0, 1, 1 }, /* 0 */ { 0, 1, 1, 1 }, { 0, 2, 1, 1 }, { 0, 3, 1, 1 }, { 0, 100, 1, 0 }, { 0, 100, 1, 1 }, { 0, 100, 1, 5 }, { 0, 100, 1, 10 }, { 0, 100, 1, 50 }, { 0, 100, 1, 100 }, { 0, 100, 1, 150 }, /* 10 */ { 0, 100, 3, 0 }, { 0, 100, 3, 1 }, { 0, 100, 3, 5 }, { 0, 100, 3, 10 }, { 0, 100, 3, 50 }, { 0, 100, 3, 100 }, { 0, 100, 3, 150 }, { 0, 100, 5, 1 }, { 0, 100, -3, 0 }, { 0, 100, -3, 1 }, /* 20 */ { 0, 100, -3, 5 }, { 0, 100, -3, 10 }, { 0, 100, -3, 50 }, { 0, 100, -3, 100 }, { 0, 100, -3, 150 }, { 0, 100, 10, 1 }, { 0, 100, 50, 1 }, { 0, 100, 100, 1 }, { 0, 100, 150, 1 }, { 0, 0x10000000, 1, 123 }, /* 30 */ { 0, 0x20000000, 1, 123 }, { 0, 0x40000000, 1, 123 }, { 0, -0x80000000, 1, 123 }, { 50, 50, 1, 1 }, { 50, 50, 1, 2 }, { 50, 50, 1, -1 }, { 50, 50, 1, -2 }, { 50, 50, 2, 1 }, { 50, 50, 3, 1 }, { 100, 200, 3, 1 }, /* 40 */ { 100, 200, 3, -1 }, { 0x7ffffffe, -0x80000000, 1, 123 }, { 0x7fffffff, -0x80000000, 1, 123 }, }; int num_threads = pomp_get_num_threads(); int thread_num = pomp_get_thread_num(); int i; for (i = 0; i < sizeof(tests)/sizeof(tests[0]); i++) { int my_begin, my_end, my_next, my_lastchunk; int begin, end, next, lastchunk; unsigned int my_loops, loops; DWORD broken_flags; my_loops = my_begin = my_end = my_next = my_lastchunk = 0xdeadbeef; loops = begin = end = next = lastchunk = 0xdeadbeef; broken_flags = my_for_static_init(tests[i].first, tests[i].last, tests[i].step, tests[i].chunksize, &my_loops, &my_begin, &my_end, &my_next, &my_lastchunk); p_vcomp_for_static_init(tests[i].first, tests[i].last, tests[i].step, tests[i].chunksize, &loops, &begin, &end, &next, &lastchunk); if (broken_flags & VCOMP_FOR_STATIC_BROKEN_LOOP) { ok(loops == 0 || loops == 1, "test %d, thread %d/%d: expected loops == 0 or 1, got %u\n", i, thread_num, num_threads, loops); } else { ok(loops == my_loops, "test %d, thread %d/%d: expected loops == %u, got %u\n", i, thread_num, num_threads, my_loops, loops); ok(begin == my_begin, "test %d, thread %d/%d: expected begin == %d, got %d\n", i, thread_num, num_threads, my_begin, begin); ok(end == my_end, "test %d, thread %d/%d: expected end == %d, got %d\n", i, thread_num, num_threads, my_end, end); ok(next == my_next || broken(broken_flags & VCOMP_FOR_STATIC_BROKEN_NEXT), "test %d, thread %d/%d: expected next == %d, got %d\n", i, thread_num, num_threads, my_next, next); ok(lastchunk == my_lastchunk, "test %d, thread %d/%d: expected lastchunk == %d, got %d\n", i, thread_num, num_threads, my_lastchunk, lastchunk); } p_vcomp_for_static_end(); p_vcomp_barrier(); if (tests[i].first == tests[i].last) continue; my_loops = my_begin = my_end = my_next = my_lastchunk = 0xdeadbeef; loops = begin = end = next = lastchunk = 0xdeadbeef; broken_flags = my_for_static_init(tests[i].last, tests[i].first, tests[i].step, tests[i].chunksize, &my_loops, &my_begin, &my_end, &my_next, &my_lastchunk); p_vcomp_for_static_init(tests[i].last, tests[i].first, tests[i].step, tests[i].chunksize, &loops, &begin, &end, &next, &lastchunk); if (broken_flags & VCOMP_FOR_STATIC_BROKEN_LOOP) { ok(loops == 0 || loops == 1, "test %d, thread %d/%d: expected loops == 0 or 1, got %u\n", i, thread_num, num_threads, loops); } else { ok(loops == my_loops, "test %d, thread %d/%d: expected loops == %u, got %u\n", i, thread_num, num_threads, my_loops, loops); ok(begin == my_begin, "test %d, thread %d/%d: expected begin == %d, got %d\n", i, thread_num, num_threads, my_begin, begin); ok(end == my_end, "test %d, thread %d/%d: expected end == %d, got %d\n", i, thread_num, num_threads, my_end, end); ok(next == my_next || broken(broken_flags & VCOMP_FOR_STATIC_BROKEN_NEXT), "test %d, thread %d/%d: expected next == %d, got %d\n", i, thread_num, num_threads, my_next, next); ok(lastchunk == my_lastchunk, "test %d, thread %d/%d: expected lastchunk == %d, got %d\n", i, thread_num, num_threads, my_lastchunk, lastchunk); } p_vcomp_for_static_end(); p_vcomp_barrier(); } } #undef VCOMP_FOR_STATIC_BROKEN_LOOP #undef VCOMP_FOR_STATIC_BROKEN_NEXT static void test_vcomp_for_static_init(void) { int max_threads = pomp_get_max_threads(); int i; for_static_cb(); for (i = 1; i <= 4; i++) { pomp_set_num_threads(i); p_vcomp_fork(TRUE, 0, for_static_cb); p_vcomp_fork(FALSE, 0, for_static_cb); } pomp_set_num_threads(max_threads); } static void CDECL for_dynamic_static_cb(void) { unsigned int my_begin, my_end, begin, end; int ret; begin = end = 0xdeadbeef; my_for_static_simple_init(TRUE, 0, 1000, 7, TRUE, &my_begin, &my_end); p_vcomp_for_dynamic_init(VCOMP_DYNAMIC_FLAGS_STATIC | VCOMP_DYNAMIC_FLAGS_INCREMENT, 0, 1000, 7, 1); ret = p_vcomp_for_dynamic_next(&begin, &end); ok(ret == TRUE, "expected ret == TRUE, got %d\n", ret); ok(begin == my_begin, "expected begin == %u, got %u\n", my_begin, begin); ok(end == my_end, "expected end == %u, got %u\n", my_end, end); ret = p_vcomp_for_dynamic_next(&begin, &end); ok(ret == FALSE, "expected ret == FALSE, got %d\n", ret); begin = end = 0xdeadbeef; my_for_static_simple_init(TRUE, 1000, 0, 7, FALSE, &my_begin, &my_end); p_vcomp_for_dynamic_init(VCOMP_DYNAMIC_FLAGS_STATIC, 1000, 0, 7, 1); ret = p_vcomp_for_dynamic_next(&begin, &end); ok(ret == TRUE, "expected ret == TRUE, got %d\n", ret); ok(begin == my_begin, "expected begin == %u, got %u\n", my_begin, begin); ok(end == my_end, "expected end == %u, got %u\n", my_end, end); ret = p_vcomp_for_dynamic_next(&begin, &end); ok(ret == FALSE, "expected ret == FALSE, got %d\n", ret); begin = end = 0xdeadbeef; my_for_static_simple_init(TRUE, 0, 1000, 7, TRUE, &my_begin, &my_end); p_vcomp_for_dynamic_init(VCOMP_DYNAMIC_FLAGS_STATIC | VCOMP_DYNAMIC_FLAGS_INCREMENT, 0, 1000, 7, 5); ret = p_vcomp_for_dynamic_next(&begin, &end); ok(ret == TRUE, "expected ret == TRUE, got %d\n", ret); ok(begin == my_begin, "expected begin == %u, got %u\n", my_begin, begin); ok(end == my_end, "expected end == %u, got %u\n", my_end, end); ret = p_vcomp_for_dynamic_next(&begin, &end); ok(ret == FALSE, "expected ret == FALSE, got %d\n", ret); begin = end = 0xdeadbeef; my_for_static_simple_init(TRUE, 1000, 0, 7, FALSE, &my_begin, &my_end); p_vcomp_for_dynamic_init(VCOMP_DYNAMIC_FLAGS_STATIC, 1000, 0, 7, 5); ret = p_vcomp_for_dynamic_next(&begin, &end); ok(ret == TRUE, "expected ret == TRUE, got %d\n", ret); ok(begin == my_begin, "expected begin == %u, got %u\n", my_begin, begin); ok(end == my_end, "expected end == %u, got %u\n", my_end, end); ret = p_vcomp_for_dynamic_next(&begin, &end); ok(ret == FALSE, "expected ret == FALSE, got %d\n", ret); } static void CDECL for_dynamic_chunked_cb(LONG *a, LONG *b, LONG *c, LONG *d) { unsigned int begin, end; p_vcomp_for_dynamic_init(VCOMP_DYNAMIC_FLAGS_CHUNKED | VCOMP_DYNAMIC_FLAGS_INCREMENT, 0, 1000, 7, 1); while (p_vcomp_for_dynamic_next(&begin, &end)) { if (begin == 994) ok(end == 1000, "expected end == 1000, got %u\n", end); else ok(begin == end, "expected begin == end, got %u and %u\n", begin, end); InterlockedExchangeAdd(a, begin); } p_vcomp_for_dynamic_init(VCOMP_DYNAMIC_FLAGS_CHUNKED, 1000, 0, 7, 1); while (p_vcomp_for_dynamic_next(&begin, &end)) { if (begin == 6) ok(end == 0, "expected end == 0, got %u\n", end); else ok(begin == end, "expected begin == end, got %u and %u\n", begin, end); InterlockedExchangeAdd(b, begin); } p_vcomp_for_dynamic_init(VCOMP_DYNAMIC_FLAGS_CHUNKED | VCOMP_DYNAMIC_FLAGS_INCREMENT, 0, 1000, 7, 5); while (p_vcomp_for_dynamic_next(&begin, &end)) { if (begin == 980) ok(end == 1000, "expected end == 1000, got %u\n", end); else ok(begin + 28 == end, "expected begin + 28 == end, got %u and %u\n", begin + 28, end); InterlockedExchangeAdd(c, begin); } p_vcomp_for_dynamic_init(VCOMP_DYNAMIC_FLAGS_CHUNKED, 1000, 0, 7, 5); while (p_vcomp_for_dynamic_next(&begin, &end)) { if (begin == 20) ok(end == 0, "expected end == 0, got %u\n", end); else ok(begin - 28 == end, "expected begin - 28 == end, got %u and %u\n", begin - 28, end); InterlockedExchangeAdd(d, begin); } } static void CDECL for_dynamic_guided_cb(unsigned int flags, LONG *a, LONG *b, LONG *c, LONG *d) { int num_threads = pomp_get_num_threads(); unsigned int begin, end; p_vcomp_for_dynamic_init(flags | VCOMP_DYNAMIC_FLAGS_INCREMENT, 0, 1000, 7, 1); while (p_vcomp_for_dynamic_next(&begin, &end)) { ok(num_threads != 1 || (begin == 0 && end == 1000), "expected begin == 0 and end == 1000, got %u and %u\n", begin, end); InterlockedExchangeAdd(a, begin); } p_vcomp_for_dynamic_init(flags, 1000, 0, 7, 1); while (p_vcomp_for_dynamic_next(&begin, &end)) { ok(num_threads != 1 || (begin == 1000 && end == 0), "expected begin == 1000 and end == 0, got %u and %u\n", begin, end); InterlockedExchangeAdd(b, begin); } p_vcomp_for_dynamic_init(flags | VCOMP_DYNAMIC_FLAGS_INCREMENT, 0, 1000, 7, 5); while (p_vcomp_for_dynamic_next(&begin, &end)) { ok(num_threads != 1 || (begin == 0 && end == 1000), "expected begin == 0 and end == 1000, got %u and %u\n", begin, end); InterlockedExchangeAdd(c, begin); } p_vcomp_for_dynamic_init(flags, 1000, 0, 7, 5); while (p_vcomp_for_dynamic_next(&begin, &end)) { ok(num_threads != 1 || (begin == 1000 && end == 0), "expected begin == 1000 and end == 0, got %u and %u\n", begin, end); InterlockedExchangeAdd(d, begin); } } static void test_vcomp_for_dynamic_init(void) { static const int guided_a[] = {0, 6041, 9072, 11179}; static const int guided_b[] = {1000, 1959, 2928, 3821}; static const int guided_c[] = {0, 4067, 6139, 7273}; static const int guided_d[] = {1000, 1933, 2861, 3727}; LONG a, b, c, d; int max_threads = pomp_get_max_threads(); int i; /* test static scheduling */ for_dynamic_static_cb(); for (i = 1; i <= 4; i++) { pomp_set_num_threads(i); p_vcomp_fork(TRUE, 0, for_dynamic_static_cb); p_vcomp_fork(FALSE, 0, for_dynamic_static_cb); } /* test chunked scheduling */ a = b = c = d = 0; for_dynamic_chunked_cb(&a, &b, &c, &d); ok(a == 71071, "expected a == 71071, got %d\n", a); ok(b == 71929, "expected b == 71929, got %d\n", b); ok(c == 14210, "expected c == 14210, got %d\n", c); ok(d == 14790, "expected d == 14790, got %d\n", d); for (i = 1; i <= 4; i++) { pomp_set_num_threads(i); a = b = c = d = 0; p_vcomp_fork(TRUE, 4, for_dynamic_chunked_cb, &a, &b, &c, &d); ok(a == 71071, "expected a == 71071, got %d\n", a); ok(b == 71929, "expected b == 71929, got %d\n", b); ok(c == 14210, "expected c == 14210, got %d\n", c); ok(d == 14790, "expected d == 14790, got %d\n", d); a = b = c = d = 0; p_vcomp_fork(FALSE, 4, for_dynamic_chunked_cb, &a, &b, &c, &d); ok(a == 71071, "expected a == 71071, got %d\n", a); ok(b == 71929, "expected b == 71929, got %d\n", b); ok(c == 14210, "expected c == 14210, got %d\n", c); ok(d == 14790, "expected d == 14790, got %d\n", d); } /* test guided scheduling */ a = b = c = d = 0; for_dynamic_guided_cb(VCOMP_DYNAMIC_FLAGS_GUIDED, &a, &b, &c, &d); ok(a == guided_a[0], "expected a == %d, got %d\n", guided_a[0], a); ok(b == guided_b[0], "expected b == %d, got %d\n", guided_b[0], b); ok(c == guided_c[0], "expected c == %d, got %d\n", guided_c[0], c); ok(d == guided_d[0], "expected d == %d, got %d\n", guided_d[0], d); for (i = 1; i <= 4; i++) { pomp_set_num_threads(i); a = b = c = d = 0; p_vcomp_fork(TRUE, 5, for_dynamic_guided_cb, VCOMP_DYNAMIC_FLAGS_GUIDED, &a, &b, &c, &d); ok(a == guided_a[i - 1], "expected a == %d, got %d\n", guided_a[i - 1], a); ok(b == guided_b[i - 1], "expected b == %d, got %d\n", guided_b[i - 1], b); ok(c == guided_c[i - 1], "expected c == %d, got %d\n", guided_c[i - 1], c); ok(d == guided_d[i - 1], "expected d == %d, got %d\n", guided_d[i - 1], d); a = b = c = d = 0; p_vcomp_fork(FALSE, 5, for_dynamic_guided_cb, VCOMP_DYNAMIC_FLAGS_GUIDED, &a, &b, &c, &d); ok(a == guided_a[0], "expected a == %d, got %d\n", guided_a[0], a); ok(b == guided_b[0], "expected b == %d, got %d\n", guided_b[0], b); ok(c == guided_c[0], "expected c == %d, got %d\n", guided_c[0], c); ok(d == guided_d[0], "expected d == %d, got %d\n", guided_d[0], d); } /* test with empty flags */ a = b = c = d = 0; for_dynamic_guided_cb(0, &a, &b, &c, &d); ok(a == guided_a[0], "expected a == %d, got %d\n", guided_a[0], a); ok(b == guided_b[0], "expected b == %d, got %d\n", guided_b[0], b); ok(c == guided_c[0], "expected c == %d, got %d\n", guided_c[0], c); ok(d == guided_d[0], "expected d == %d, got %d\n", guided_d[0], d); for (i = 1; i <= 4; i++) { pomp_set_num_threads(i); a = b = c = d = 0; p_vcomp_fork(TRUE, 5, for_dynamic_guided_cb, 0, &a, &b, &c, &d); ok(a == guided_a[i - 1], "expected a == %d, got %d\n", guided_a[i - 1], a); ok(b == guided_b[i - 1], "expected b == %d, got %d\n", guided_b[i - 1], b); ok(c == guided_c[i - 1], "expected c == %d, got %d\n", guided_c[i - 1], c); ok(d == guided_d[i - 1], "expected d == %d, got %d\n", guided_d[i - 1], d); a = b = c = d = 0; p_vcomp_fork(FALSE, 5, for_dynamic_guided_cb, 0, &a, &b, &c, &d); ok(a == guided_a[0], "expected a == %d, got %d\n", guided_a[0], a); ok(b == guided_b[0], "expected b == %d, got %d\n", guided_b[0], b); ok(c == guided_c[0], "expected c == %d, got %d\n", guided_c[0], c); ok(d == guided_d[0], "expected d == %d, got %d\n", guided_d[0], d); } pomp_set_num_threads(max_threads); } static void CDECL master_cb(HANDLE semaphore) { int num_threads = pomp_get_num_threads(); int thread_num = pomp_get_thread_num(); if (p_vcomp_master_begin()) { ok(thread_num == 0, "expected thread_num == 0, got %d\n", thread_num); if (num_threads >= 2) { DWORD result = WaitForSingleObject(semaphore, 1000); ok(result == WAIT_OBJECT_0, "WaitForSingleObject returned %u\n", result); } p_vcomp_master_end(); } if (thread_num == 1) ReleaseSemaphore(semaphore, 1, NULL); } static void test_vcomp_master_begin(void) { int max_threads = pomp_get_max_threads(); HANDLE semaphore; int i; semaphore = CreateSemaphoreA(NULL, 0, 1, NULL); ok(semaphore != NULL, "CreateSemaphoreA failed %u\n", GetLastError()); master_cb(semaphore); for (i = 1; i <= 4; i++) { pomp_set_num_threads(i); p_vcomp_fork(TRUE, 1, master_cb, semaphore); p_vcomp_fork(FALSE, 1, master_cb, semaphore); } CloseHandle(semaphore); pomp_set_num_threads(max_threads); } static void CDECL single_cb(int flags, HANDLE semaphore) { int num_threads = pomp_get_num_threads(); if (p_vcomp_single_begin(flags)) { if (num_threads >= 2) { DWORD result = WaitForSingleObject(semaphore, 1000); ok(result == WAIT_OBJECT_0, "WaitForSingleObject returned %u\n", result); } } p_vcomp_single_end(); if (p_vcomp_single_begin(flags)) { if (num_threads >= 2) ReleaseSemaphore(semaphore, 1, NULL); } p_vcomp_single_end(); } static void test_vcomp_single_begin(void) { int max_threads = pomp_get_max_threads(); HANDLE semaphore; int i; semaphore = CreateSemaphoreA(NULL, 0, 1, NULL); ok(semaphore != NULL, "CreateSemaphoreA failed %u\n", GetLastError()); single_cb(0, semaphore); single_cb(1, semaphore); for (i = 1; i <= 4; i++) { pomp_set_num_threads(i); p_vcomp_fork(TRUE, 2, single_cb, 0, semaphore); p_vcomp_fork(TRUE, 2, single_cb, 1, semaphore); p_vcomp_fork(FALSE, 2, single_cb, 0, semaphore); p_vcomp_fork(FALSE, 2, single_cb, 1, semaphore); } CloseHandle(semaphore); pomp_set_num_threads(max_threads); } static void CDECL critsect_cb(LONG *a) { static CRITICAL_SECTION *critsect; LONG tmp; p_vcomp_enter_critsect(&critsect); tmp = *a; Sleep(50); *a = tmp + 1; p_vcomp_leave_critsect(critsect); ok(critsect != NULL, "expected critsect != NULL\n"); EnterCriticalSection(critsect); tmp = *a; Sleep(50); *a = tmp + 1; LeaveCriticalSection(critsect); } static void test_vcomp_enter_critsect(void) { int max_threads = pomp_get_max_threads(); LONG a; int i; a = 0; critsect_cb(&a); ok(a == 2, "expected a == 2, got %d\n", a); for (i = 1; i <= 4; i++) { pomp_set_num_threads(i); a = 0; p_vcomp_fork(TRUE, 1, critsect_cb, &a); ok(a == 2 * i, "expected a == %d, got %d\n", 2 * i, a); a = 0; p_vcomp_fork(FALSE, 1, critsect_cb, &a); ok(a == 2, "expected a == 2, got %d\n", a); } pomp_set_num_threads(max_threads); } static void test_vcomp_flush(void) { p_vcomp_flush(); p_vcomp_flush(); p_vcomp_flush(); } static void test_omp_init_lock(void) { omp_lock_t lock; int ret; pomp_init_lock(&lock); /* test omp_set_lock */ pomp_set_lock(&lock); pomp_unset_lock(&lock); /* test omp_test_lock */ ret = pomp_test_lock(&lock); ok(ret == 1, "expected ret == 1, got %d\n", ret); ret = pomp_test_lock(&lock); ok(ret == 0, "expected ret == 0, got %d\n", ret); pomp_unset_lock(&lock); /* test with EnterCriticalSection */ EnterCriticalSection(lock); ret = pomp_test_lock(&lock); todo_wine ok(ret == 1, "expected ret == 1, got %d\n", ret); if (ret) { ret = pomp_test_lock(&lock); ok(ret == 0, "expected ret == 0, got %d\n", ret); pomp_unset_lock(&lock); } LeaveCriticalSection(lock); pomp_destroy_lock(&lock); } static void test_omp_init_nest_lock(void) { omp_nest_lock_t lock; int ret; ok(pomp_init_nest_lock == pomp_init_lock, "expected omp_init_nest_lock == %p, got %p\n", pomp_init_lock, pomp_init_nest_lock); ok(pomp_destroy_nest_lock == pomp_destroy_lock, "expected omp_destroy_nest_lock == %p, got %p\n", pomp_destroy_lock, pomp_destroy_nest_lock); pomp_init_nest_lock(&lock); /* test omp_set_nest_lock */ pomp_set_nest_lock(&lock); pomp_set_nest_lock(&lock); pomp_unset_nest_lock(&lock); pomp_unset_nest_lock(&lock); /* test omp_test_nest_lock */ ret = pomp_test_nest_lock(&lock); ok(ret == 1, "expected ret == 1, got %d\n", ret); ret = pomp_test_nest_lock(&lock); ok(ret == 2, "expected ret == 2, got %d\n", ret); ret = pomp_test_nest_lock(&lock); ok(ret == 3, "expected ret == 3, got %d\n", ret); pomp_unset_nest_lock(&lock); pomp_unset_nest_lock(&lock); pomp_unset_nest_lock(&lock); /* test with EnterCriticalSection */ EnterCriticalSection(lock); ret = pomp_test_nest_lock(&lock); todo_wine ok(ret == 1, "expected ret == 1, got %d\n", ret); pomp_unset_nest_lock(&lock); LeaveCriticalSection(lock); pomp_destroy_nest_lock(&lock); } static void test_atomic_integer32(void) { struct { void (CDECL *func)(int *, int); int v1, v2, expected; } tests1[] = { { p_vcomp_atomic_add_i4, 0x11223344, 0x77665544, -0x77777778 }, { p_vcomp_atomic_and_i4, 0x11223344, 0x77665544, 0x11221144 }, { p_vcomp_atomic_div_i4, 0x77665544, 0x11223344, 6 }, { p_vcomp_atomic_div_i4, 0x77665544, -0x11223344, -6 }, { p_vcomp_atomic_mul_i4, 0x11223344, 0x77665544, -0xecccdf0 }, { p_vcomp_atomic_mul_i4, 0x11223344, -0x77665544, 0xecccdf0 }, { p_vcomp_atomic_or_i4, 0x11223344, 0x77665544, 0x77667744 }, { p_vcomp_atomic_shl_i4, 0x11223344, 3, -0x76ee65e0 }, { p_vcomp_atomic_shl_i4, -0x11223344, 3, 0x76ee65e0 }, { p_vcomp_atomic_shr_i4, 0x11223344, 3, 0x2244668 }, { p_vcomp_atomic_shr_i4, -0x11223344, 3, -0x2244669 }, { p_vcomp_atomic_sub_i4, 0x11223344, 0x77665544, -0x66442200 }, { p_vcomp_atomic_xor_i4, 0x11223344, 0x77665544, 0x66446600 }, #if defined(__i386__) || defined(__x86_64__) { p_vcomp_atomic_shl_i4, 0x11223344, 35, -0x76ee65e0 }, { p_vcomp_atomic_shr_i4, 0x11223344, 35, 0x2244668 }, #endif }; struct { void (CDECL *func)(unsigned int *, unsigned int); unsigned int v1, v2, expected; } tests2[] = { { p_vcomp_atomic_div_ui4, 0x77665544, 0x11223344, 6 }, { p_vcomp_atomic_div_ui4, 0x77665544, 0xeeddccbc, 0 }, { p_vcomp_atomic_shr_ui4, 0x11223344, 3, 0x2244668 }, { p_vcomp_atomic_shr_ui4, 0xeeddccbc, 3, 0x1ddbb997 }, #if defined(__i386__) || defined(__x86_64__) { p_vcomp_atomic_shr_ui4, 0x11223344, 35, 0x2244668 }, #endif }; int i; for (i = 0; i < sizeof(tests1)/sizeof(tests1[0]); i++) { int val = tests1[i].v1; tests1[i].func(&val, tests1[i].v2); ok(val == tests1[i].expected, "test %d: expected val == %d, got %d\n", i, tests1[i].expected, val); } for (i = 0; i < sizeof(tests2)/sizeof(tests2[0]); i++) { unsigned int val = tests2[i].v1; tests2[i].func(&val, tests2[i].v2); ok(val == tests2[i].expected, "test %d: expected val == %u, got %u\n", i, tests2[i].expected, val); } } static void test_atomic_integer64(void) { struct { void (CDECL *func)(LONG64 *, LONG64); LONG64 v1, v2, expected; } tests1[] = { { p_vcomp_atomic_add_i8, 0x1122334455667788, 0x7766554433221100, -0x7777777777777778 }, { p_vcomp_atomic_and_i8, 0x1122334455667788, 0x7766554433221100, 0x1122114411221100 }, { p_vcomp_atomic_div_i8, 0x7766554433221100, 0x1122334455667788, 6 }, { p_vcomp_atomic_div_i8, 0x7766554433221100, -0x1122334455667788, -6 }, { p_vcomp_atomic_mul_i8, 0x1122334455667788, 0x7766554433221100, 0x3e963337c6000800 }, { p_vcomp_atomic_mul_i8, 0x1122334455667788, -0x7766554433221100, 0xc169ccc839fff800 }, { p_vcomp_atomic_or_i8, 0x1122334455667788, 0x7766554433221100, 0x7766774477667788 }, { p_vcomp_atomic_sub_i8, 0x1122334455667788, 0x7766554433221100, -0x664421ffddbb9978 }, { p_vcomp_atomic_xor_i8, 0x1122334455667788, 0x7766554433221100, 0x6644660066446688 }, }; struct { void (CDECL *func)(LONG64 *, unsigned int); LONG64 v1; unsigned int v2; LONG64 expected; BOOL todo; } tests2[] = { { p_vcomp_atomic_shl_i8, 0x1122334455667788, 3, -0x76ee65dd54cc43c0 }, { p_vcomp_atomic_shl_i8, 0x1122334455667788, 60, 0x8000000000000000 }, { p_vcomp_atomic_shl_i8, -0x1122334455667788, 3, 0x76ee65dd54cc43c0 }, { p_vcomp_atomic_shr_i8, 0x1122334455667788, 3, 0x22446688aaccef1 }, { p_vcomp_atomic_shr_i8, 0x1122334455667788, 60, 1 }, { p_vcomp_atomic_shr_i8, -0x1122334455667788, 3, -0x22446688aaccef1 }, #if defined(__i386__) { p_vcomp_atomic_shl_i8, 0x1122334455667788, 64, 0, TRUE }, { p_vcomp_atomic_shl_i8, 0x1122334455667788, 67, 0, TRUE }, { p_vcomp_atomic_shr_i8, 0x1122334455667788, 64, 0, TRUE }, { p_vcomp_atomic_shr_i8, 0x1122334455667788, 67, 0, TRUE }, #elif defined(__x86_64__) { p_vcomp_atomic_shl_i8, 0x1122334455667788, 64, 0x1122334455667788 }, { p_vcomp_atomic_shl_i8, 0x1122334455667788, 67, -0x76ee65dd54cc43c0 }, { p_vcomp_atomic_shr_i8, 0x1122334455667788, 64, 0x1122334455667788 }, { p_vcomp_atomic_shr_i8, 0x1122334455667788, 67, 0x22446688aaccef1 }, #endif }; struct { void (CDECL *func)(ULONG64 *, ULONG64); ULONG64 v1, v2, expected; } tests3[] = { { p_vcomp_atomic_div_ui8, 0x7766554455667788, 0x1122334433221100, 6 }, { p_vcomp_atomic_div_ui8, 0x7766554455667788, 0xeeddccbbaa998878, 0 }, }; struct { void (CDECL *func)(ULONG64 *, unsigned int); ULONG64 v1; unsigned int v2; ULONG64 expected; BOOL todo; } tests4[] = { { p_vcomp_atomic_shr_ui8, 0x1122334455667788, 3, 0x22446688aaccef1 }, { p_vcomp_atomic_shr_ui8, 0x1122334455667788, 60, 1 }, { p_vcomp_atomic_shr_ui8, 0xeeddccbbaa998878, 3, 0x1ddbb9977553310f }, #if defined(__i386__) { p_vcomp_atomic_shr_ui8, 0x1122334455667788, 64, 0, TRUE }, { p_vcomp_atomic_shr_ui8, 0x1122334455667788, 67, 0, TRUE }, #elif defined(__x86_64__) { p_vcomp_atomic_shr_ui8, 0x1122334455667788, 64, 0x1122334455667788 }, { p_vcomp_atomic_shr_ui8, 0x1122334455667788, 67, 0x22446688aaccef1 }, #endif }; int i; for (i = 0; i < sizeof(tests1)/sizeof(tests1[0]); i++) { LONG64 val = tests1[i].v1; tests1[i].func(&val, tests1[i].v2); ok(val == tests1[i].expected, "test %d: unexpectedly got %s\n", i, debugstr_longlong(val)); } for (i = 0; i < sizeof(tests2)/sizeof(tests2[0]); i++) { LONG64 val = tests2[i].v1; tests2[i].func(&val, tests2[i].v2); todo_wine_if(tests2[i].todo) ok(val == tests2[i].expected, "test %d: unexpectedly got %s\n", i, debugstr_longlong(val)); } for (i = 0; i < sizeof(tests3)/sizeof(tests3[0]); i++) { ULONG64 val = tests3[i].v1; tests3[i].func(&val, tests3[i].v2); ok(val == tests3[i].expected, "test %d: unexpectedly got %s\n", i, debugstr_longlong(val)); } for (i = 0; i < sizeof(tests4)/sizeof(tests4[0]); i++) { ULONG64 val = tests4[i].v1; tests4[i].func(&val, tests4[i].v2); todo_wine_if(tests4[i].todo) ok(val == tests4[i].expected, "test %d: unexpectedly got %s\n", i, debugstr_longlong(val)); } } static void test_atomic_float(void) { struct { void (CDECL *func)(float *, float); float v1, v2, expected; } tests[] = { { p_vcomp_atomic_add_r4, 42.0, 17.0, 42.0 + 17.0 }, { p_vcomp_atomic_div_r4, 42.0, 17.0, 42.0 / 17.0 }, { p_vcomp_atomic_mul_r4, 42.0, 17.0, 42.0 * 17.0 }, { p_vcomp_atomic_sub_r4, 42.0, 17.0, 42.0 - 17.0 }, }; int i; for (i = 0; i < sizeof(tests)/sizeof(tests[0]); i++) { float val = tests[i].v1; tests[i].func(&val, tests[i].v2); ok(tests[i].expected - 0.001 < val && val < tests[i].expected + 0.001, "test %d: expected val == %f, got %f\n", i, tests[i].expected, val); } } static void test_atomic_double(void) { struct { void (CDECL *func)(double *, double); double v1, v2, expected; } tests[] = { { p_vcomp_atomic_add_r8, 42.0, 17.0, 42.0 + 17.0 }, { p_vcomp_atomic_div_r8, 42.0, 17.0, 42.0 / 17.0 }, { p_vcomp_atomic_mul_r8, 42.0, 17.0, 42.0 * 17.0 }, { p_vcomp_atomic_sub_r8, 42.0, 17.0, 42.0 - 17.0 }, }; int i; for (i = 0; i < sizeof(tests)/sizeof(tests[0]); i++) { double val = tests[i].v1; tests[i].func(&val, tests[i].v2); ok(tests[i].expected - 0.001 < val && val < tests[i].expected + 0.001, "test %d: expected val == %f, got %f\n", i, tests[i].expected, val); } } START_TEST(vcomp) { if (!init_vcomp()) return; test_omp_get_num_threads(FALSE); test_omp_get_num_threads(TRUE); test_vcomp_fork(); test_vcomp_sections_init(); test_vcomp_for_static_simple_init(); test_vcomp_for_static_init(); test_vcomp_for_dynamic_init(); test_vcomp_master_begin(); test_vcomp_single_begin(); test_vcomp_enter_critsect(); test_vcomp_flush(); test_omp_init_lock(); test_omp_init_nest_lock(); test_atomic_integer32(); test_atomic_integer64(); test_atomic_float(); test_atomic_double(); release_vcomp(); }