Mercurial > vec
view test/test_arith.h @ 45:7955bed1d169 default tip
*: add preliminary floating point support
no x86 intrinsics just yet, but I did add altivec since it's
(arguably) the simplest :)
author | Paper <paper@tflc.us> |
---|---|
date | Wed, 30 Apr 2025 18:36:38 -0400 |
parents | c6e0df09b86f |
children |
line wrap: on
line source
#define CREATE_TEST_EX(type, ctype, print, bits, size, op, equiv, secondtype, secondctype) \ static int test_arith_v##type##bits##x##size##_##op(v##type##bits##x##size a, v##secondtype##bits##x##size b) \ { \ V##ctype##bits##x##size##_ALIGNED_ARRAY(orig_a); \ V##secondctype##bits##x##size##_ALIGNED_ARRAY(orig_b); \ V##ctype##bits##x##size##_ALIGNED_ARRAY(orig_c); \ \ v##type##bits##x##size c = v##type##bits##x##size##_##op(a, b); \ \ v##type##bits##x##size##_store_aligned(a, orig_a); \ v##secondtype##bits##x##size##_store_aligned(b, orig_b); \ v##type##bits##x##size##_store_aligned(c, orig_c); \ \ for (int i = 0; i < size; i++) { \ if ((vec_##type##bits)(equiv) != orig_c[i]) { \ fprintf(stderr, "v" #type #bits "x" #size "_" #op " test FAILED at index %d: (%s) [%" print "] does not equal result [%" print "]!\n", i, #equiv, (vec_##type##bits)(equiv), orig_c[i]); \ print_v##type##bits##x##size(stderr,a); \ print_v##secondtype##bits##x##size(stderr,b); \ print_v##type##bits##x##size(stderr,c); \ fprintf(stderr, "\n"); \ return 1; \ } \ } \ \ return 0; \ } #define CREATE_TEST(type, ctype, print, bits, size, op, equiv) \ CREATE_TEST_EX(type, ctype, print, bits, size, op, equiv, type, ctype) #define CREATE_TEST_SHIFT(type, ctype, print, bits, size, op, equiv) \ CREATE_TEST_EX(type, ctype, print, bits, size, op, equiv, uint, UINT) #define CREATE_TESTS_INT(type, ctype, sign, print, bits, size) \ CREATE_TEST(type, ctype, print, bits, size, add, orig_a[i] + orig_b[i]) \ CREATE_TEST(type, ctype, print, bits, size, sub, orig_a[i] - orig_b[i]) \ CREATE_TEST(type, ctype, print, bits, size, mul, orig_a[i] * orig_b[i]) \ CREATE_TEST(type, ctype, print, bits, size, div, (orig_b[i]) ? (orig_a[i] / orig_b[i]) : 0) \ CREATE_TEST(type, ctype, print, bits, size, mod, (orig_b[i]) ? (orig_a[i] % orig_b[i]) : 0) \ CREATE_TEST(type, ctype, print, bits, size, and, orig_a[i] & orig_b[i]) \ CREATE_TEST(type, ctype, print, bits, size, or, orig_a[i] | orig_b[i]) \ CREATE_TEST(type, ctype, print, bits, size, xor, orig_a[i] ^ orig_b[i]) \ CREATE_TEST(type, ctype, print, bits, size, avg, (vec_##type##bits)vec_im##sign##avg(orig_a[i], orig_b[i])) \ CREATE_TEST_SHIFT(type, ctype, print, bits, size, rshift, vec_##sign##rshift(orig_a[i], orig_b[i])) \ CREATE_TEST_SHIFT(type, ctype, print, bits, size, lshift, vec_##sign##lshift(orig_a[i], orig_b[i])) \ CREATE_TEST_SHIFT(type, ctype, print, bits, size, lrshift, vec_urshift((vec_uint##bits)orig_a[i], orig_b[i])) \ CREATE_TEST(type, ctype, print, bits, size, min, (orig_a[i] < orig_b[i]) ? orig_a[i] : orig_b[i]) \ CREATE_TEST(type, ctype, print, bits, size, max, (orig_a[i] > orig_b[i]) ? orig_a[i] : orig_b[i]) #define CREATE_TESTS_FLOAT(bits, size) \ CREATE_TEST(f, F, "f", bits, size, add, orig_a[i] + orig_b[i]) \ CREATE_TEST(f, F, "f", bits, size, sub, orig_a[i] - orig_b[i]) \ CREATE_TEST(f, F, "f", bits, size, mul, orig_a[i] * orig_b[i]) \ CREATE_TEST(f, F, "f", bits, size, div, (orig_b[i]) ? (orig_a[i] / orig_b[i]) : 0) \ CREATE_TEST(f, F, "f", bits, size, mod, (orig_b[i]) ? (fmod(orig_a[i], orig_b[i])) : 0) \ CREATE_TEST(f, F, "f", bits, size, avg, (orig_a[i] + orig_b[i]) / 2) \ CREATE_TEST(f, F, "f", bits, size, min, (orig_a[i] < orig_b[i]) ? orig_a[i] : orig_b[i]) \ CREATE_TEST(f, F, "f", bits, size, max, (orig_a[i] > orig_b[i]) ? orig_a[i] : orig_b[i]) #define CREATE_TESTS(bits, size) \ CREATE_TESTS_INT(int, INT, /* nothing */, PRI##d##bits, bits, size) \ CREATE_TESTS_INT(uint, UINT, u, PRI##u##bits, bits, size) CREATE_TESTS(8, 2) CREATE_TESTS(8, 4) CREATE_TESTS(16, 2) CREATE_TESTS(8, 8) CREATE_TESTS(16, 4) CREATE_TESTS(32, 2) CREATE_TESTS(8, 16) CREATE_TESTS(16, 8) CREATE_TESTS(32, 4) CREATE_TESTS(64, 2) CREATE_TESTS(8, 32) CREATE_TESTS(16, 16) CREATE_TESTS(32, 8) CREATE_TESTS(64, 4) CREATE_TESTS(8, 64) CREATE_TESTS(16, 32) CREATE_TESTS(32, 16) CREATE_TESTS(64, 8) CREATE_TESTS_FLOAT(32, 2) CREATE_TESTS_FLOAT(32, 4) CREATE_TESTS_FLOAT(32, 8) CREATE_TESTS_FLOAT(32, 16) CREATE_TESTS_FLOAT(64, 2) CREATE_TESTS_FLOAT(64, 4) CREATE_TESTS_FLOAT(64, 8) #undef CREATE_TESTS_INT #undef CREATE_TESTS_FLOAT #undef CREATE_TESTS #undef CREATE_TEST #undef CREATE_TEST_SHIFT static int test_arith(void) { int ret = 0; #define RUN_TESTS_SIGN(shorttype, type, bits, size) \ for (size_t i = 0U; i < ARRAY_SIZE(testval##shorttype##bits); i++) { \ const v##type##bits##x##size a = vtest##shorttype##bits##x##size(i); \ for (size_t j = 0U; j < ARRAY_SIZE(testval##shorttype##bits); j++) { \ const v##type##bits##x##size b = vtest##shorttype##bits##x##size(j); \ ret |= test_arith_v##type##bits##x##size##_add(a, b); \ ret |= test_arith_v##type##bits##x##size##_sub(a, b); \ ret |= test_arith_v##type##bits##x##size##_mul(a, b); \ ret |= test_arith_v##type##bits##x##size##_div(a, b); \ ret |= test_arith_v##type##bits##x##size##_mod(a, b); \ ret |= test_arith_v##type##bits##x##size##_and(a, b); \ ret |= test_arith_v##type##bits##x##size##_or(a, b); \ ret |= test_arith_v##type##bits##x##size##_xor(a, b); \ ret |= test_arith_v##type##bits##x##size##_avg(a, b); \ ret |= test_arith_v##type##bits##x##size##_min(a, b); \ ret |= test_arith_v##type##bits##x##size##_max(a, b); \ } \ } \ \ for (size_t i = 0U; i < ARRAY_SIZE(testval##shorttype##bits); i++) { \ const v##type##bits##x##size a = vtest##shorttype##bits##x##size(i); \ for (uint32_t j = 0U; j < bits; j++) { \ const vuint##bits##x##size b = vuint##bits##x##size##_splat(j); \ ret |= test_arith_v##type##bits##x##size##_rshift(a, b); \ ret |= test_arith_v##type##bits##x##size##_lshift(a, b); \ ret |= test_arith_v##type##bits##x##size##_lrshift(a, b); \ } \ } #define RUN_TESTS(bits, size) \ RUN_TESTS_SIGN( , int, bits, size) \ RUN_TESTS_SIGN(u, uint, bits, size) #define RUN_TESTS_FLOAT(shorttype, type, bits, size) \ for (size_t i = 0U; i < ARRAY_SIZE(testval##shorttype##bits); i++) { \ const v##type##bits##x##size a = vtest##shorttype##bits##x##size(i); \ for (size_t j = 0U; j < ARRAY_SIZE(testval##shorttype##bits); j++) { \ const v##type##bits##x##size b = vtest##shorttype##bits##x##size(j); \ ret |= test_arith_v##type##bits##x##size##_add(a, b); \ ret |= test_arith_v##type##bits##x##size##_sub(a, b); \ ret |= test_arith_v##type##bits##x##size##_mul(a, b); \ ret |= test_arith_v##type##bits##x##size##_div(a, b); \ ret |= test_arith_v##type##bits##x##size##_mod(a, b); \ ret |= test_arith_v##type##bits##x##size##_avg(a, b); \ ret |= test_arith_v##type##bits##x##size##_min(a, b); \ ret |= test_arith_v##type##bits##x##size##_max(a, b); \ } \ } RUN_TESTS(8, 2) RUN_TESTS(8, 4) RUN_TESTS(16, 2) RUN_TESTS(8, 8) RUN_TESTS(16, 4) RUN_TESTS(32, 2) RUN_TESTS(8, 16) RUN_TESTS(16, 8) RUN_TESTS(32, 4) RUN_TESTS(64, 2) RUN_TESTS(8, 32) RUN_TESTS(16, 16) RUN_TESTS(32, 8) RUN_TESTS(64, 4) RUN_TESTS(8, 64) RUN_TESTS(16, 32) RUN_TESTS(32, 16) RUN_TESTS(64, 8) RUN_TESTS_FLOAT(f, f, 32, 2) RUN_TESTS_FLOAT(f, f, 32, 4) RUN_TESTS_FLOAT(f, f, 32, 8) RUN_TESTS_FLOAT(f, f, 32, 16) RUN_TESTS_FLOAT(f, f, 64, 2) RUN_TESTS_FLOAT(f, f, 64, 4) RUN_TESTS_FLOAT(f, f, 64, 8) #undef RUN_TESTS_FLOAT #undef RUN_TESTS_SIGN #undef RUN_TESTS return ret; }