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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
|
/* Function atan2 vectorized with SSE4.
Copyright (C) 2021 Free Software Foundation, Inc.
This file is part of the GNU C Library.
The GNU C Library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public
License as published by the Free Software Foundation; either
version 2.1 of the License, or (at your option) any later version.
The GNU C Library is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public
License along with the GNU C Library; if not, see
https://www.gnu.org/licenses/. */
/*
* ALGORITHM DESCRIPTION:
* For 0.0 <= x <= 7.0/16.0: atan(x) = atan(0.0) + atan(s), where s=(x-0.0)/(1.0+0.0*x)
* For 7.0/16.0 <= x <= 11.0/16.0: atan(x) = atan(0.5) + atan(s), where s=(x-0.5)/(1.0+0.5*x)
* For 11.0/16.0 <= x <= 19.0/16.0: atan(x) = atan(1.0) + atan(s), where s=(x-1.0)/(1.0+1.0*x)
* For 19.0/16.0 <= x <= 39.0/16.0: atan(x) = atan(1.5) + atan(s), where s=(x-1.5)/(1.0+1.5*x)
* For 39.0/16.0 <= x <= inf : atan(x) = atan(inf) + atan(s), where s=-1.0/x
* Where atan(s) ~= s+s^3*Poly11(s^2) on interval |s|<7.0/0.16.
*
*
*/
/* Offsets for data table __svml_datan2_data_internal
*/
#define dPI 0
#define dPIO2 16
#define dA19 32
#define dA18 48
#define dA17 64
#define dA16 80
#define dA15 96
#define dA14 112
#define dA13 128
#define dA12 144
#define dA11 160
#define dA10 176
#define dA09 192
#define dA08 208
#define dA07 224
#define dA06 240
#define dA05 256
#define dA04 272
#define dA03 288
#define dA02 304
#define dA01 320
#define dA00 336
#define dSIGN_MASK 352
#define iCHK_WORK_SUB 368
#define iCHK_WORK_CMP 384
#define dABS_MASK 400
#define dZERO 416
#include <sysdep.h>
.text
.section .text.sse4,"ax",@progbits
ENTRY(_ZGVbN2vv_atan2_sse4)
subq $88, %rsp
cfi_def_cfa_offset(96)
movaps %xmm0, %xmm8
/*
* #define NO_VECTOR_ZERO_ATAN2_ARGS
* Declarations
* Variables
* Constants
* The end of declarations
* Implementation
* Get r0~=1/B
* Cannot be replaced by VQRCP(D, dR0, dB);
* Argument Absolute values
*/
movups dABS_MASK+__svml_datan2_data_internal(%rip), %xmm4
movaps %xmm1, %xmm9
movaps %xmm4, %xmm1
andps %xmm8, %xmm4
andps %xmm9, %xmm1
movaps %xmm4, %xmm2
cmpnltpd %xmm1, %xmm2
/* Argument signs */
movups dSIGN_MASK+__svml_datan2_data_internal(%rip), %xmm3
movaps %xmm2, %xmm0
movups dPIO2+__svml_datan2_data_internal(%rip), %xmm5
movaps %xmm3, %xmm7
movaps %xmm3, %xmm6
/*
* 1) If y<x then a= y, b=x, PIO2=0
* 2) If y>x then a=-x, b=y, PIO2=Pi/2
*/
orps %xmm1, %xmm3
movaps %xmm2, %xmm10
andps %xmm2, %xmm5
andnps %xmm4, %xmm0
andps %xmm2, %xmm3
andnps %xmm1, %xmm10
andps %xmm4, %xmm2
orps %xmm3, %xmm0
orps %xmm2, %xmm10
divpd %xmm10, %xmm0
movq iCHK_WORK_SUB+__svml_datan2_data_internal(%rip), %xmm11
/* if x<0, dPI = Pi, else dPI =0 */
movaps %xmm9, %xmm3
/* Check if y and x are on main path. */
pshufd $221, %xmm1, %xmm12
andps %xmm9, %xmm7
psubd %xmm11, %xmm12
andps %xmm8, %xmm6
movq iCHK_WORK_CMP+__svml_datan2_data_internal(%rip), %xmm13
xorl %edx, %edx
movups %xmm4, 16(%rsp)
xorl %eax, %eax
pshufd $221, %xmm4, %xmm14
movdqa %xmm12, %xmm4
pcmpgtd %xmm13, %xmm4
pcmpeqd %xmm13, %xmm12
por %xmm12, %xmm4
/* Polynomial. */
movaps %xmm0, %xmm12
mulpd %xmm0, %xmm12
cmplepd dZERO+__svml_datan2_data_internal(%rip), %xmm3
psubd %xmm11, %xmm14
movdqa %xmm14, %xmm15
pcmpeqd %xmm13, %xmm14
pcmpgtd %xmm13, %xmm15
por %xmm14, %xmm15
movaps %xmm12, %xmm14
mulpd %xmm12, %xmm14
por %xmm15, %xmm4
movaps %xmm14, %xmm15
mulpd %xmm14, %xmm15
movmskps %xmm4, %ecx
movups %xmm10, (%rsp)
movups dA19+__svml_datan2_data_internal(%rip), %xmm10
mulpd %xmm15, %xmm10
movups dA18+__svml_datan2_data_internal(%rip), %xmm13
movups dA17+__svml_datan2_data_internal(%rip), %xmm11
addpd dA15+__svml_datan2_data_internal(%rip), %xmm10
mulpd %xmm15, %xmm13
mulpd %xmm15, %xmm11
mulpd %xmm15, %xmm10
addpd dA14+__svml_datan2_data_internal(%rip), %xmm13
addpd dA13+__svml_datan2_data_internal(%rip), %xmm11
addpd dA11+__svml_datan2_data_internal(%rip), %xmm10
mulpd %xmm15, %xmm13
mulpd %xmm15, %xmm11
mulpd %xmm15, %xmm10
addpd dA10+__svml_datan2_data_internal(%rip), %xmm13
addpd dA09+__svml_datan2_data_internal(%rip), %xmm11
addpd dA07+__svml_datan2_data_internal(%rip), %xmm10
mulpd %xmm15, %xmm13
mulpd %xmm15, %xmm11
mulpd %xmm15, %xmm10
addpd dA06+__svml_datan2_data_internal(%rip), %xmm13
addpd dA05+__svml_datan2_data_internal(%rip), %xmm11
addpd dA03+__svml_datan2_data_internal(%rip), %xmm10
mulpd %xmm15, %xmm13
mulpd %xmm15, %xmm11
mulpd %xmm12, %xmm10
addpd dA02+__svml_datan2_data_internal(%rip), %xmm13
addpd dA01+__svml_datan2_data_internal(%rip), %xmm11
addpd %xmm10, %xmm13
mulpd %xmm11, %xmm12
mulpd %xmm13, %xmm14
movups dA16+__svml_datan2_data_internal(%rip), %xmm2
mulpd %xmm15, %xmm2
addpd dA12+__svml_datan2_data_internal(%rip), %xmm2
mulpd %xmm15, %xmm2
addpd dA08+__svml_datan2_data_internal(%rip), %xmm2
mulpd %xmm15, %xmm2
addpd dA04+__svml_datan2_data_internal(%rip), %xmm2
/* A00=1.0, account for it later VQFMA(D, dP4, dP4, dR8, dA00); */
mulpd %xmm2, %xmm15
addpd %xmm12, %xmm15
addpd %xmm14, %xmm15
/*
* Reconstruction.
* dP=(R+R*dP) + dPIO2
*/
mulpd %xmm0, %xmm15
addpd %xmm15, %xmm0
addpd %xmm5, %xmm0
andps __svml_datan2_data_internal(%rip), %xmm3
orps %xmm7, %xmm0
addpd %xmm3, %xmm0
/* Special branch for fast (vector) processing of zero arguments */
movups 16(%rsp), %xmm11
orps %xmm6, %xmm0
testb $3, %cl
/* Go to auxilary branch */
jne L(AUX_BRANCH)
# LOE rbx rbp r12 r13 r14 r15 eax edx xmm0 xmm1 xmm3 xmm4 xmm5 xmm6 xmm7 xmm8 xmm9 xmm11
/* Return from auxilary branch
* for out of main path inputs
*/
L(AUX_BRANCH_RETURN):
/*
* Special branch for fast (vector) processing of zero arguments
* The end of implementation
*/
testl %edx, %edx
/* Go to special inputs processing branch */
jne L(SPECIAL_VALUES_BRANCH)
# LOE rbx rbp r12 r13 r14 r15 eax edx xmm0 xmm8 xmm9
/* Restore registers
* and exit the function
*/
L(EXIT):
addq $88, %rsp
cfi_def_cfa_offset(8)
ret
cfi_def_cfa_offset(96)
/* Branch to process
* special inputs
*/
L(SPECIAL_VALUES_BRANCH):
movups %xmm8, 32(%rsp)
movups %xmm9, 48(%rsp)
movups %xmm0, 64(%rsp)
# LOE rbx rbp r12 r13 r14 r15 eax edx xmm0
movq %r12, 16(%rsp)
cfi_offset(12, -80)
movl %eax, %r12d
movq %r13, 8(%rsp)
cfi_offset(13, -88)
movl %edx, %r13d
movq %r14, (%rsp)
cfi_offset(14, -96)
# LOE rbx rbp r15 r12d r13d
/* Range mask
* bits check
*/
L(RANGEMASK_CHECK):
btl %r12d, %r13d
/* Call scalar math function */
jc L(SCALAR_MATH_CALL)
# LOE rbx rbp r15 r12d r13d
/* Special inputs
* processing loop
*/
L(SPECIAL_VALUES_LOOP):
incl %r12d
cmpl $2, %r12d
/* Check bits in range mask */
jl L(RANGEMASK_CHECK)
# LOE rbx rbp r15 r12d r13d
movq 16(%rsp), %r12
cfi_restore(12)
movq 8(%rsp), %r13
cfi_restore(13)
movq (%rsp), %r14
cfi_restore(14)
movups 64(%rsp), %xmm0
/* Go to exit */
jmp L(EXIT)
cfi_offset(12, -80)
cfi_offset(13, -88)
cfi_offset(14, -96)
# LOE rbx rbp r12 r13 r14 r15 xmm0
/* Scalar math fucntion call
* to process special input
*/
L(SCALAR_MATH_CALL):
movl %r12d, %r14d
movsd 32(%rsp,%r14,8), %xmm0
movsd 48(%rsp,%r14,8), %xmm1
call atan2@PLT
# LOE rbx rbp r14 r15 r12d r13d xmm0
movsd %xmm0, 64(%rsp,%r14,8)
/* Process special inputs in loop */
jmp L(SPECIAL_VALUES_LOOP)
cfi_restore(12)
cfi_restore(13)
cfi_restore(14)
# LOE rbx rbp r15 r12d r13d
/* Auxilary branch
* for out of main path inputs
*/
L(AUX_BRANCH):
/* Check if at least on of Y or Y is zero: iAXAYZERO */
movups dZERO+__svml_datan2_data_internal(%rip), %xmm2
/* Check if both X & Y are not NaNs: iXYnotNAN */
movaps %xmm9, %xmm12
movaps %xmm8, %xmm10
cmpordpd %xmm9, %xmm12
cmpordpd %xmm8, %xmm10
cmpeqpd %xmm2, %xmm1
cmpeqpd %xmm2, %xmm11
andps %xmm10, %xmm12
orps %xmm11, %xmm1
pshufd $221, %xmm1, %xmm1
pshufd $221, %xmm12, %xmm11
/* Check if at least on of Y or Y is zero and not NaN: iAXAYZEROnotNAN */
pand %xmm11, %xmm1
/* Exclude from previous callout mask zero (and not NaN) arguments */
movdqa %xmm1, %xmm13
pandn %xmm4, %xmm13
/*
* Path for zero arguments (at least one of both)
* Check if both args are zeros (den. is zero)
*/
movups (%rsp), %xmm4
cmpeqpd %xmm2, %xmm4
/* Go to callout */
movmskps %xmm13, %edx
/* Set sPIO2 to zero if den. is zero */
movaps %xmm4, %xmm15
andps %xmm2, %xmm4
andnps %xmm5, %xmm15
andl $3, %edx
orps %xmm4, %xmm15
pshufd $221, %xmm9, %xmm5
orps %xmm7, %xmm15
/* Res = sign(Y)*(X<0)?(PIO2+PI):PIO2 */
pshufd $221, %xmm2, %xmm7
pcmpgtd %xmm5, %xmm7
pshufd $80, %xmm7, %xmm14
andps %xmm3, %xmm14
addpd %xmm14, %xmm15
/* Merge results from main and spec path */
pshufd $80, %xmm1, %xmm3
orps %xmm6, %xmm15
movdqa %xmm3, %xmm6
andps %xmm3, %xmm15
andnps %xmm0, %xmm6
movaps %xmm6, %xmm0
orps %xmm15, %xmm0
/* Return to main vector processing path */
jmp L(AUX_BRANCH_RETURN)
# LOE rbx rbp r12 r13 r14 r15 eax edx xmm0 xmm8 xmm9
END(_ZGVbN2vv_atan2_sse4)
.section .rodata, "a"
.align 16
#ifdef __svml_datan2_data_internal_typedef
typedef unsigned int VUINT32;
typedef struct {
__declspec(align(16)) VUINT32 dPI[2][2];
__declspec(align(16)) VUINT32 dPIO2[2][2];
__declspec(align(16)) VUINT32 dA19[2][2];
__declspec(align(16)) VUINT32 dA18[2][2];
__declspec(align(16)) VUINT32 dA17[2][2];
__declspec(align(16)) VUINT32 dA16[2][2];
__declspec(align(16)) VUINT32 dA15[2][2];
__declspec(align(16)) VUINT32 dA14[2][2];
__declspec(align(16)) VUINT32 dA13[2][2];
__declspec(align(16)) VUINT32 dA12[2][2];
__declspec(align(16)) VUINT32 dA11[2][2];
__declspec(align(16)) VUINT32 dA10[2][2];
__declspec(align(16)) VUINT32 dA09[2][2];
__declspec(align(16)) VUINT32 dA08[2][2];
__declspec(align(16)) VUINT32 dA07[2][2];
__declspec(align(16)) VUINT32 dA06[2][2];
__declspec(align(16)) VUINT32 dA05[2][2];
__declspec(align(16)) VUINT32 dA04[2][2];
__declspec(align(16)) VUINT32 dA03[2][2];
__declspec(align(16)) VUINT32 dA02[2][2];
__declspec(align(16)) VUINT32 dA01[2][2];
__declspec(align(16)) VUINT32 dA00[2][2];
__declspec(align(16)) VUINT32 dSIGN_MASK[2][2];
__declspec(align(16)) VUINT32 iCHK_WORK_SUB[4][1];
__declspec(align(16)) VUINT32 iCHK_WORK_CMP[4][1];
__declspec(align(16)) VUINT32 dABS_MASK[2][2];
__declspec(align(16)) VUINT32 dZERO[2][2];
} __svml_datan2_data_internal;
#endif
__svml_datan2_data_internal:
.quad 0x400921FB54442D18, 0x400921FB54442D18 //dPI
.align 16
.quad 0x3FF921FB54442D18, 0x3FF921FB54442D18 //dPIO2
.align 16
.quad 0xBEF4FDB537ABC7A3, 0xBEF4FDB537ABC7A3 // dA19
.align 16
.quad 0x3F2CED0A36665209, 0x3F2CED0A36665209 // dA18
.align 16
.quad 0xBF52E67C93954C23, 0xBF52E67C93954C23 // dA17
.align 16
.quad 0x3F6F5A1DAE82AFB3, 0x3F6F5A1DAE82AFB3 // dA16
.align 16
.quad 0xBF82B2EC618E4BAD, 0xBF82B2EC618E4BAD // dA15
.align 16
.quad 0x3F914F4C661116A5, 0x3F914F4C661116A5 // dA14
.align 16
.quad 0xBF9A5E83B081F69C, 0xBF9A5E83B081F69C // dA13
.align 16
.quad 0x3FA169980CB6AD4F, 0x3FA169980CB6AD4F // dA12
.align 16
.quad 0xBFA4EFA2E563C1BC, 0xBFA4EFA2E563C1BC // dA11
.align 16
.quad 0x3FA7EC0FBC50683B, 0x3FA7EC0FBC50683B // dA10
.align 16
.quad 0xBFAAD261EAA09954, 0xBFAAD261EAA09954 // dA09
.align 16
.quad 0x3FAE1749BD612DCF, 0x3FAE1749BD612DCF // dA08
.align 16
.quad 0xBFB11084009435E0, 0xBFB11084009435E0 // dA07
.align 16
.quad 0x3FB3B12A49295651, 0x3FB3B12A49295651 // dA06
.align 16
.quad 0xBFB745D009BADA94, 0xBFB745D009BADA94 // dA05
.align 16
.quad 0x3FBC71C707F7D5B5, 0x3FBC71C707F7D5B5 // dA04
.align 16
.quad 0xBFC2492491EE55C7, 0xBFC2492491EE55C7 // dA03
.align 16
.quad 0x3FC999999997EE34, 0x3FC999999997EE34 // dA02
.align 16
.quad 0xBFD55555555553C5, 0xBFD55555555553C5 // dA01
.align 16
.quad 0x3FF0000000000000, 0x3FF0000000000000 // dA00
.align 16
.quad 0x8000000000000000, 0x8000000000000000 //dSIGN_MASK
.align 16
.long 0x80300000, 0x80300000, 0x80300000, 0x80300000 //iCHK_WORK_SUB
.align 16
.long 0xfdd00000, 0xfdd00000, 0xfdd00000, 0xfdd00000 //iCHK_WORK_CMP
.align 16
.quad 0x7fffffffffffffff, 0x7fffffffffffffff //dABS_MASK
.align 16
.quad 0x0000000000000000, 0x0000000000000000 //dZERO
.align 16
.type __svml_datan2_data_internal,@object
.size __svml_datan2_data_internal,.-__svml_datan2_data_internal
|
