1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
/*
* Test various integer multiply intrinsics.
* Here we check for _mm512_[mask]mul_ep[i|u]32 intrinsics.
*/
#include "m512_test_util.h"
#include <stdio.h>
#include <string.h>
volatile int vol0 = 0;
V512 i64;
V512 i64_mix;
V512 i64_big;
void NOINLINE init() {
volatile int i;
for (i = 0; i < 8; i++) {
i64.s64[i] = i;
i64_mix.s64[i] = (i & 1) ? i : -i;
i64_big.s64[i] = 1000 * (i + 1);
if ((i & 1) != 0) {
i64_big.s64[i] = -i64_big.s64[i];
}
}
}
void NOINLINE do_muldq() {
V512 res;
V512 expected;
__mmask16 k;
volatile int i;
res.zmmi = _mm512_mul_epi32(i64_mix.zmmi, i64_big.zmmi);
for (i = 0; i < 8; i++) {
expected.s64[i] = (I64)i64_mix.s32[2 * i] * i64_big.s32[2 * i];
}
check_equal_nd(&res, &expected, 16, "_mm512_mul_epi32", __LINE__);
/*
* No-op to inhibit PRE of i64_big, thus enabling localized ciscization.
*/
i64_big.xmm[vol0] = i64_big.xmm[vol0];
k = 0xcd;
res.zmmi = _mm512_setzero_epi32();
res.zmmi = _mm512_mask_mul_epi32(res.zmmi, k, i64.zmmi, i64_big.zmmi);
for (i = 0; i < 8; i++) {
expected.s64[i] = 0;
if ((k & (1 << i)) != 0) {
expected.s64[i] = (I64)i64.s32[2 * i] * i64_big.s32[2 * i];
}
}
check_equal_nd(&res, &expected, 16, "_mm512_mask_mul_epi32", __LINE__);
}
void NOINLINE do_muludq() {
V512 res;
V512 expected;
__mmask16 k;
volatile int i;
res.zmmi = _mm512_mul_epu32(i64_mix.zmmi, i64_big.zmmi);
for (i = 0; i < 8; i++) {
expected.u64[i] = (U64)i64_mix.u32[2 * i] * i64_big.u32[2 * i];
}
check_equal_nd(&res, &expected, 16, "_mm512_mul_epu32", __LINE__);
/*
* No-op to inhibit PRE of i64_big, thus enabling localized ciscization.
*/
i64_big.xmm[vol0] = i64_big.xmm[vol0];
k = 0xcd;
res.zmmi = _mm512_setzero_epi32();
res.zmmi = _mm512_mask_mul_epu32(res.zmmi, k, i64.zmmi, i64_big.zmmi);
for (i = 0; i < 8; i++) {
expected.u64[i] = 0;
if ((k & (1 << i)) != 0) {
expected.u64[i] = (U64)i64.u32[2 * i] * i64_big.u32[2 * i];
}
}
check_equal_nd(&res, &expected, 16, "_mm512_mask_mul_epu32", __LINE__);
}
int main(int argc, char *argv[]) {
init();
do_muldq();
do_muludq();
if (n_errs != 0) {
printf("FAILED\n");
return 1;
}
printf("PASSED\n");
return 0;
}
|
