01 Feb 2010
 

4x4 matrix multiplication with SSE (GCC inline asm)

 
 
 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
/* It's two times faster than pure C code. */
void matrix44_mult_SSE(float* r, const float* a, const float* b)
{
asm("movaps 0x00(%1),%%xmm0\n"
"movaps 0x10(%1),%%xmm1\n"
"movaps 0x20(%1),%%xmm2\n"
"movaps 0x30(%1),%%xmm3\n"
// 0, 1, 2, 3
"movaps 0x00(%2),%%xmm4\n"
"movaps %%xmm4,%%xmm5\n"
"movaps %%xmm4,%%xmm6\n"
"movaps %%xmm4,%%xmm7\n"
"shufps $0x00,%%xmm4,%%xmm4\n"
"shufps $0x55,%%xmm5,%%xmm5\n"
"shufps $0xAA,%%xmm6,%%xmm6\n"
"shufps $0xFF,%%xmm7,%%xmm7\n"
"mulps %%xmm0,%%xmm4\n"
"mulps %%xmm1,%%xmm5\n"
"mulps %%xmm2,%%xmm6\n"
"mulps %%xmm3,%%xmm7\n"
"addps %%xmm5,%%xmm4\n"
"addps %%xmm6,%%xmm4\n"
"addps %%xmm7,%%xmm4\n"
"movaps %%xmm4,0x00(%0)\n"
// 4, 5, 6, 7
"movaps 0x10(%2),%%xmm4\n"
"movaps %%xmm4,%%xmm5\n"
"movaps %%xmm4,%%xmm6\n"
"movaps %%xmm4,%%xmm7\n"
"shufps $0x00,%%xmm4,%%xmm4\n"
"shufps $0x55,%%xmm5,%%xmm5\n"
"shufps $0xAA,%%xmm6,%%xmm6\n"
"shufps $0xFF,%%xmm7,%%xmm7\n"
"mulps %%xmm0,%%xmm4\n"
"mulps %%xmm1,%%xmm5\n"
"mulps %%xmm2,%%xmm6\n"
"mulps %%xmm3,%%xmm7\n"
"addps %%xmm5,%%xmm4\n"
"addps %%xmm6,%%xmm4\n"
"addps %%xmm7,%%xmm4\n"
"movaps %%xmm4,0x10(%0)\n"
// 8, 9, 10, 11
"movaps 0x20(%2),%%xmm4\n"
"movaps %%xmm4,%%xmm5\n"
"movaps %%xmm4,%%xmm6\n"
"movaps %%xmm4,%%xmm7\n"
"shufps $0x00,%%xmm4,%%xmm4\n"
"shufps $0x55,%%xmm5,%%xmm5\n"
"shufps $0xAA,%%xmm6,%%xmm6\n"
"shufps $0xFF,%%xmm7,%%xmm7\n"
"mulps %%xmm0,%%xmm4\n"
"mulps %%xmm1,%%xmm5\n"
"mulps %%xmm2,%%xmm6\n"
"mulps %%xmm3,%%xmm7\n"
"addps %%xmm5,%%xmm4\n"
"addps %%xmm6,%%xmm4\n"
"addps %%xmm7,%%xmm4\n"
"movaps %%xmm4,0x20(%0)\n"
// 12, 13, 14, 15
"movaps 0x30(%2),%%xmm4\n"
"movaps %%xmm4,%%xmm5\n"
"movaps %%xmm4,%%xmm6\n"
"movaps %%xmm4,%%xmm7\n"
"shufps $0x00,%%xmm4,%%xmm4\n"
"shufps $0x55,%%xmm5,%%xmm5\n"
"shufps $0xAA,%%xmm6,%%xmm6\n"
"shufps $0xFF,%%xmm7,%%xmm7\n"
"mulps %%xmm0,%%xmm4\n"
"mulps %%xmm1,%%xmm5\n"
"mulps %%xmm2,%%xmm6\n"
"mulps %%xmm3,%%xmm7\n"
"addps %%xmm5,%%xmm4\n"
"addps %%xmm6,%%xmm4\n"
"addps %%xmm7,%%xmm4\n"
"movaps %%xmm4,0x30(%0)\n"
:
: "r" (r), "r" (a), "r" (b)
: "xmm0", "xmm1", "xmm2", "xmm3",
"xmm4", "xmm5", "xmm6", "xmm7"
);
}