blob: a19a13f1d3082540b589515c8a348574ffad4f0e (
plain)
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
104
105
106
107
108
|
#include <stdint.h>
#ifdef LASX
#define ELEM_COUNT 32
#define REPLGR2VR_B "xvreplgr2vr.b $xr"
#define ST "xvst $xr"
#else
#define ELEM_COUNT 16
#define REPLGR2VR_B "vreplgr2vr.b $vr"
#define ST "vst $vr"
#endif
// base is added to each value. If base = 2, then
// assume the vector element type is char:
// $reg0 = { 0x02 * $ELEM_COUNT }
// $reg1 = { 0x03 * $ELEM_COUNT } etc.
void write_simd_regs(unsigned base) {
#define WRITE_SIMD(NUM) \
asm volatile(REPLGR2VR_B #NUM ", %0\n\t" ::"r"(base + NUM))
WRITE_SIMD(0);
WRITE_SIMD(1);
WRITE_SIMD(2);
WRITE_SIMD(3);
WRITE_SIMD(4);
WRITE_SIMD(5);
WRITE_SIMD(6);
WRITE_SIMD(7);
WRITE_SIMD(8);
WRITE_SIMD(9);
WRITE_SIMD(10);
WRITE_SIMD(11);
WRITE_SIMD(12);
WRITE_SIMD(13);
WRITE_SIMD(14);
WRITE_SIMD(15);
WRITE_SIMD(16);
WRITE_SIMD(17);
WRITE_SIMD(18);
WRITE_SIMD(19);
WRITE_SIMD(20);
WRITE_SIMD(21);
WRITE_SIMD(22);
WRITE_SIMD(23);
WRITE_SIMD(24);
WRITE_SIMD(25);
WRITE_SIMD(26);
WRITE_SIMD(27);
WRITE_SIMD(28);
WRITE_SIMD(29);
WRITE_SIMD(30);
WRITE_SIMD(31);
}
unsigned verify_simd_regs() {
uint8_t simd_reg[ELEM_COUNT];
uint8_t target = 0;
#define VERIFY_SIMD(NUM) \
do { \
for (int i = 0; i < ELEM_COUNT; ++i) \
simd_reg[i] = 0; \
asm volatile(ST #NUM ", %0\n\t" ::"m"(simd_reg)); \
target = NUM + 1; \
for (int i = 0; i < ELEM_COUNT; ++i) \
if (simd_reg[i] != target) \
return 1; \
} while (0)
VERIFY_SIMD(0);
VERIFY_SIMD(1);
VERIFY_SIMD(2);
VERIFY_SIMD(3);
VERIFY_SIMD(4);
VERIFY_SIMD(5);
VERIFY_SIMD(6);
VERIFY_SIMD(7);
VERIFY_SIMD(8);
VERIFY_SIMD(9);
VERIFY_SIMD(10);
VERIFY_SIMD(11);
VERIFY_SIMD(12);
VERIFY_SIMD(13);
VERIFY_SIMD(14);
VERIFY_SIMD(15);
VERIFY_SIMD(16);
VERIFY_SIMD(17);
VERIFY_SIMD(18);
VERIFY_SIMD(19);
VERIFY_SIMD(20);
VERIFY_SIMD(21);
VERIFY_SIMD(22);
VERIFY_SIMD(23);
VERIFY_SIMD(24);
VERIFY_SIMD(25);
VERIFY_SIMD(26);
VERIFY_SIMD(27);
VERIFY_SIMD(28);
VERIFY_SIMD(29);
VERIFY_SIMD(30);
VERIFY_SIMD(31);
return 0;
}
int main(int argc, char *argv[]) {
write_simd_regs(0);
return verify_simd_regs(); // Set break point at this line.
}
|
