diff options
Diffstat (limited to 'SingleSource/UnitTests/Vector/AVX512F/m512_op_ps.c')
-rw-r--r-- | SingleSource/UnitTests/Vector/AVX512F/m512_op_ps.c | 236 |
1 files changed, 236 insertions, 0 deletions
diff --git a/SingleSource/UnitTests/Vector/AVX512F/m512_op_ps.c b/SingleSource/UnitTests/Vector/AVX512F/m512_op_ps.c new file mode 100644 index 00000000..59d8763a --- /dev/null +++ b/SingleSource/UnitTests/Vector/AVX512F/m512_op_ps.c @@ -0,0 +1,236 @@ +#include "m512_test_util.h" +#include <memory.h> +#include <stdio.h> +#include <stdlib.h> + +/* + * This test was created to check the correctness + * of the following intrinsics support: + * _mm512_add_ps() + * _mm512_max_ps() + * _mm512_min_ps() + * _mm512_mask_max_ps() + * _mm512_mask_min_ps() + * _mm512_mask_mul_ps() + * _mm512_mask_abs_ps() + * _mm512_add_round_ps() + * _mm512_sub_round_ps() + */ + + +int show_op = 0; + +typedef enum { ASSIGN, ABS, ADD, MAX, MIN, MUL, SUB } OPER; + +static void NOINLINE intop(OPER op, float ivalout[16], float ivalop1[16], + float ivalop2[16]) { + int i; + int handled = 0; + + memset(ivalout, 0, sizeof(ivalout)); + for (i = 0; i < 16; i += 1) { + switch (op) { + case ASSIGN: + handled = 1; + ivalout[i] = ivalop1[i]; + break; + case ADD: + handled = 1; + ivalout[i] = ivalop1[i] + ivalop2[i]; + break; + case ABS: + handled = 1; + ivalout[i] = ivalop1[i] >= 0 ? ivalop1[i] : -ivalop1[i]; + break; + case MAX: + handled = 1; + ivalout[i] = (ivalop1[i] > ivalop2[i]) ? ivalop1[i] : ivalop2[i]; + break; + case MIN: + handled = 1; + ivalout[i] = (ivalop1[i] < ivalop2[i]) ? ivalop1[i] : ivalop2[i]; + break; + case MUL: + handled = 1; + ivalout[i] = ivalop2[i] * ivalop1[i]; + break; + case SUB: + handled = 1; + ivalout[i] = ivalop1[i] - ivalop2[i]; + break; + default: + printf("FAIL: bad op\n"); + break; + } + } + if (!handled) { + printf("FAIL: unsupported op\n"); + n_errs++; + } +} + +static int NOINLINE check(float val1[], float good[]) { + int i; + int res = 1; + for (i = 0; i < 16; i += 1) { + if (val1[i] != good[i]) { + res = 0; + printf("FAIL: %f != %f\n", val1[i], good[i]); + } + } + return (res); +} + +static int NOINLINE check_mask(float val1[], float good[], int mask) { + int i; + int res = 1; + for (i = 0; i < 16; i += 1) { + if ((1 << i) & mask) { + if (val1[i] != good[i]) { + res = 0; + printf("FAIL: %f != %f\n", val1[i], good[i]); + } + } + } + return (res); +} + +static void NOINLINE print_vec(char *pfx, float ivec[]) { + if (pfx) { + printf("%s: ", pfx); + } + printf("%10.4f %10.4f %10.4f %10.4f ", ivec[15], ivec[14], ivec[13], + ivec[12]); + printf("%10.4f %10.4f %10.4f %10.4f ", ivec[11], ivec[10], ivec[9], ivec[8]); + printf("%10.4f %10.4f %10.4f %10.4f ", ivec[7], ivec[6], ivec[5], ivec[4]); + printf("%10.4f %10.4f %10.4f %10.4f\n", ivec[3], ivec[2], ivec[1], ivec[0]); +} + +#define DOONE(OP, FUNC) \ + { \ + int passed = 0; \ + intop(OP, good.f32, v1.f32, v2.f32); \ + vvv.zmm = FUNC(v1.zmm, v2.zmm); \ + passed = check(vvv.f32, good.f32); \ + if (!passed) { \ + printf("FAIL " #FUNC "\n"); \ + n_errs++; \ + } \ + if (!passed || show_op) { \ + print_vec("Opand1", v1.f32); \ + print_vec("Opand2", v2.f32); \ + print_vec("Scalar", good.f32); \ + print_vec("Vector", vvv.f32); \ + } \ + } + +#define DOONE_WITH_MASK(OP, FUNC, MMASK) \ + { \ + int passed = 0; \ + intop(OP, good.f32, v1.f32, v2.f32); \ + vvv.zmm = FUNC(vvv.zmm, MMASK, v1.zmm, v2.zmm); \ + passed = check_mask(vvv.f32, good.f32, MMASK); \ + if (!passed) { \ + printf("FAIL " #FUNC "\n"); \ + n_errs++; \ + } \ + if (!passed || show_op) { \ + print_vec("Opand1", v1.f32); \ + print_vec("Opand2", v2.f32); \ + print_vec("Scalar", good.f32); \ + print_vec("Vector", vvv.f32); \ + } \ + } + +#define DOONE_WITH_MASK_1OP(OP, FUNC, MMASK) \ + { \ + int passed = 0; \ + intop(OP, good.f32, v1.f32, v2.f32); \ + vvv.zmm = FUNC(vvv.zmm, MMASK, v1.zmm); \ + passed = check_mask(vvv.f32, good.f32, MMASK); \ + if (!passed) { \ + printf("FAIL " #FUNC "\n"); \ + n_errs++; \ + } \ + if (!passed || show_op) { \ + print_vec("Opand1", v1.f32); \ + print_vec("Opand2", v2.f32); \ + print_vec("Scalar", good.f32); \ + print_vec("Vector", vvv.f32); \ + } \ + } + +#define DOONE_ROUND(OP, FUNC, ROUND) \ + { \ + int passed = 0; \ + intop(OP, good.f32, v1.f32, v2.f32); \ + vvv.zmm = FUNC(v1.zmm, v2.zmm, ROUND); \ + passed = check(vvv.f32, good.f32); \ + if (!passed) { \ + printf("FAIL " #FUNC "\n"); \ + n_errs++; \ + } \ + if (!passed || show_op) { \ + print_vec("Opand1", v1.f32); \ + print_vec("Opand2", v2.f32); \ + print_vec("Scalar", good.f32); \ + print_vec("Vector", vvv.f32); \ + } \ + } + +#define DOONE_WITH_MASK_ROUND(OP, FUNC, MMASK, ROUND) \ + { \ + int passed = 0; \ + intop(OP, good.f32, v1.f32, v2.f32); \ + vvv.zmm = FUNC(vvv.zmm, MMASK, v1.zmm, v2.zmm, ROUND); \ + passed = check_mask(vvv.f32, good.f32, MMASK); \ + if (!passed) { \ + printf("FAIL " #FUNC "\n"); \ + n_errs++; \ + } \ + if (!passed || show_op) { \ + print_vec("Opand1", v1.f32); \ + print_vec("Opand2", v2.f32); \ + print_vec("Scalar", good.f32); \ + print_vec("Vector", vvv.f32); \ + } \ + } + +int main() { + float init1[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, -11, 12, 13, 14, 15, 16}; + float init2[] = {11, 12, 23, 24, -35, 36, 17, 38, + 42, -1, 33, 7, 8, 10, 11, 12}; + + V512 v1; + V512 v2; + V512 good; + V512 vvv; + + intop(ASSIGN, v1.f32, init1, 0); + intop(ASSIGN, v2.f32, init2, 0); + + // simple intrinsics + DOONE(ADD, _mm512_add_ps); + DOONE(MAX, _mm512_max_ps); + DOONE(MIN, _mm512_min_ps); + + DOONE_WITH_MASK(MAX, _mm512_mask_max_ps, 0xf01); + DOONE_WITH_MASK(MIN, _mm512_mask_min_ps, 0xf03); + DOONE_WITH_MASK(MUL, _mm512_mask_mul_ps, 0xff0); + + DOONE_WITH_MASK_1OP(ABS, _mm512_mask_abs_ps, 0xcf1); + + // intrinsics with rounding mode round + DOONE_ROUND(ADD, _mm512_add_round_ps, + _MM_FROUND_TO_NEAREST_INT | _MM_FROUND_NO_EXC); + DOONE_ROUND(SUB, _mm512_sub_round_ps, + _MM_FROUND_TO_POS_INF | _MM_FROUND_NO_EXC); + + if (n_errs != 0) { + printf("FAILED\n"); + return 1; + } + + printf("PASSED\n"); + return 0; +} |