mpir/mpn/x86_64/mode1o.as

175 lines
5.0 KiB
ActionScript

; Copyright 2000, 2001, 2002, 2003, 2004, 2005, 2006 Free Software
; Foundation, Inc.
;
; Copyright 2008 William Hart
;
; This file is part of the MPIR Library.
;
; The MPIR 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 MPIR 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 MPIR Library; see the file COPYING.LIB. If
; not, write to the Free Software Foundation, Inc., 51 Franklin Street,
; Fifth Floor, Boston, MA 02110-1301, USA.
;
; cycles/limb
; Hammer: 10
; Prescott/Nocona: 33
; mp_limb_t mpn_modexact_1_odd (mp_srcptr src, mp_size_t size,
; mp_limb_t divisor);
; mp_limb_t mpn_modexact_1c_odd (mp_srcptr src, mp_size_t size,
; mp_limb_t divisor, mp_limb_t carry);
;
; The dependent chain in the main loop is
;
; cycles
; sub rax, r8 1
; imul rax, r9 4
; mul rsi 5
; ----
; total 10
;
; The movq load from src seems to need to be scheduled back before the jz to
; achieve this speed, out-of-order execution apparently can't completely
; hide the latency otherwise.
;
; The l=src[i]-cbit step is rotated back too, since that allows us to avoid
; it for the first iteration (where there's no cbit).
;
; The code alignment used (32-byte) for the loop also seems necessary.
; Without that the non-PIC case has adcq crossing the 0x60 offset,
; apparently making it run at 11 cycles instead of 10.
;
; Not done:
;
; divq for size==1 was measured at about 79 cycles, compared to the inverse
; at about 25 cycles (both including function call overheads), so that's not
; used.
;
; Enhancements:
;
; For PIC, we shouldn't really need the GOT fetch for modlimb_invert_table,
; it'll be in rodata or text in libmpir.so and can be accessed directly %rip
; relative. This would be for small model only (something we don't
; presently detect, but which is all that gcc 3.3.3 supports), since 8-byte
; PC-relative relocations are apparently not available. Some rough
; experiments with binutils 2.13 looked worrylingly like it might come out
; with an unwanted text segment relocation though, even with ".protected".
; AMD64 mpn_modexact_1_odd -- exact division style remainder.
;
; mp_limb_t mpn_modexact_1_odd (
; mp_srcptr src, rdi
; mp_size_t size, rsi
; mp_limb_t divisor rdx
; );
; mp_limb_t mpn_modexact_1c_odd (
; mp_srcptr src, rdi
; mp_size_t size, rsi
; mp_limb_t divisor, rdx
; mp_limb_t carry rcx
; );
;
%include 'yasm_mac.inc'
BITS 64
align 32
G_EXTERN __gmp_modlimb_invert_table
GLOBAL_FUNC mpn_modexact_1_odd
mov ecx, 0 ; carry
GLOBAL_FUNC mpn_modexact_1c_odd
mov r8, rdx
shr edx, 1
%ifdef GSYM_PREFIX
%define mod_table ___gmp_modlimb_invert_table
%else
%define mod_table __gmp_modlimb_invert_table
%endif
; first use Newton's iteration to invert the divisor limb (d) using
; f(x) = 1/x - d and x[i+1] = x[i] - f(x[i]) / f'(x[i]) to give
; the iteration formula: x[i+1] = x[i] * (2 - d * x[i])
%ifdef PIC
mov r9, [mod_table wrt rip wrt ..gotpcrel]
%else
lea r9, [mod_table wrt rip]
%endif
and edx, 127
mov r10, rcx
movzx edx, byte [rdx+r9] ; inv -> rdx (8-bit approx)
mov rax, [rdi] ; first limb of numerator
lea r11, [rdi+rsi*8] ; pointer to top of src
mov rdi, r8 ; save divisor
lea ecx, [rdx+rdx]
imul rdx, rdx
neg rsi ; limb offset from top of src
imul edx, edi
sub ecx, edx ; inv -> rcx (16-bit approx)
lea edx, [rcx+rcx]
imul ecx, ecx
imul ecx, edi
sub edx, ecx ; inv -> rdx (32-bit approx)
xor ecx, ecx
lea r9, [rdx+rdx]
imul rdx, rdx
imul rdx, r8
sub r9, rdx ; inv -> r10 (64-bit approx)
mov rdx, r10 ; intial carry -> rdx
inc rsi ; adjust limb offset
jz label1
align 16
label0: ; now multiply through by inverse in loop
sub rax, rdx
adc rcx, 0
imul rax, r9
mul r8
mov rax, [r11+rsi*8]
sub rax, rcx
setc cl
inc rsi
jnz label0
label1: ; do final multiply
sub rax, rdx
adc rcx, 0
imul rax, r9
mul r8
lea rax, [rcx+rdx] ; return remainder
ret