127 lines
2.7 KiB
ArmAsm
127 lines
2.7 KiB
ArmAsm
#include "textflag.h"
|
|
|
|
// mul2 multiplicates FROM by 2, stores result in R1
|
|
// and uses R1, R2 and R3 for internal computations.
|
|
#define mul2(FROM, TO, R2, R3) \
|
|
VPSLLQ $1, FROM, TO \
|
|
VPALIGNR $8, TO, FROM, R2 \
|
|
PSRLQ $63, R2 \
|
|
MOVUPD ·x127x63(SB), R3 \
|
|
ANDPD TO, R3 \
|
|
VPUNPCKHQDQ R3, R3, R3 \
|
|
XORPD R2, TO \
|
|
XORPD R3, TO
|
|
|
|
|
|
#define mask(bit, src, tmp, to1, to2) \
|
|
MOVQ src, tmp \
|
|
SHRQ bit, tmp \
|
|
ANDQ $1, tmp \
|
|
NEGQ tmp \
|
|
MOVQ tmp, to1 \
|
|
VPBROADCASTB to1, to2
|
|
// FIXME VPBROADCASTB is AVX2 instruction
|
|
//https://software.intel.com/en-us/forums/intel-isa-extensions/topic/301461
|
|
|
|
#define mulBit(bit) \
|
|
MOVUPD X0, X8 \
|
|
MOVUPD X2, X9 \
|
|
mul2(X0, X5, X6, X7) \
|
|
VXORPD X1, X5, X0 \
|
|
mul2(X2, X5, X6, X7) \
|
|
VXORPD X3, X5, X2 \
|
|
mask(bit, CX, DX, X6, X5) \
|
|
VANDPD X0, X5, X1 \
|
|
XORPD X8, X1 \
|
|
VANDPD X2, X5, X3 \
|
|
XORPD X9, X3
|
|
|
|
|
|
// func mulBitRight(c00, c01, c10, c11 *[2]uint64, byte)
|
|
TEXT ·mulByteRight(SB),NOSPLIT,$0
|
|
MOVQ c00+0(FP), AX
|
|
MOVUPD (AX), X0
|
|
MOVQ c01+8(FP), BX
|
|
MOVUPD (BX), X1
|
|
MOVQ c10+16(FP), CX
|
|
MOVUPD (CX), X2
|
|
MOVQ c11+24(FP), DX
|
|
MOVUPD (DX), X3
|
|
MOVB e+32(FP), CX
|
|
|
|
mulBit($7)
|
|
mulBit($6)
|
|
mulBit($5)
|
|
mulBit($4)
|
|
mulBit($3)
|
|
mulBit($2)
|
|
mulBit($1)
|
|
mulBit($0)
|
|
|
|
MOVUPD X0, (AX)
|
|
MOVQ c10+16(FP), CX
|
|
MOVUPD X2, (CX)
|
|
MOVUPD X1, (BX)
|
|
MOVQ c11+24(FP), DX
|
|
MOVUPD X3, (DX)
|
|
|
|
RET
|
|
|
|
|
|
// func mulBitRight(c00, c01, c10, c11, e *[2]uint64)
|
|
TEXT ·mulBitRight(SB),NOSPLIT,$0
|
|
MOVQ c00+0(FP), AX
|
|
MOVUPD (AX), X0
|
|
MOVUPD X0, X8 // remember c00 value
|
|
MOVQ c01+8(FP), BX
|
|
MOVUPD (BX), X1
|
|
MOVQ c10+16(FP), CX
|
|
MOVUPD (CX), X2
|
|
MOVUPD X2, X9 // remember c10 value
|
|
MOVQ c11+24(FP), DX
|
|
MOVUPD (DX), X3
|
|
|
|
mul2(X0, X5, X6, X7)
|
|
VXORPD X1, X5, X0
|
|
mul2(X2, X5, X6, X7)
|
|
VXORPD X3, X5, X2
|
|
MOVQ e+32(FP), CX
|
|
MOVUPD (CX), X5
|
|
VANDPD X0, X5, X1
|
|
XORPD X8, X1
|
|
VANDPD X2, X5, X3
|
|
XORPD X9, X3
|
|
|
|
MOVUPD X0, (AX)
|
|
MOVQ c10+16(FP), CX
|
|
MOVUPD X2, (CX)
|
|
MOVUPD X1, (BX)
|
|
MOVUPD X3, (DX)
|
|
|
|
RET
|
|
|
|
// func mulBitRightx2(c00c10, c01c11 *[4]uint64, e *[2]uint64)
|
|
TEXT ·mulBitRightx2(SB),NOSPLIT,$0
|
|
MOVQ c00c10+0(FP), AX
|
|
VMOVDQA (AX), Y0
|
|
MOVQ c01c11+8(FP), BX
|
|
VMOVDQA (BX), Y8
|
|
|
|
VPSLLQ $1, Y0, Y1
|
|
VPALIGNR $8, Y1, Y0, Y2
|
|
VPSRLQ $63, Y2, Y2
|
|
VPXOR Y1, Y2, Y2
|
|
VPSRLQ $63, Y1, Y3
|
|
VPSLLQ $63, Y3, Y3
|
|
VPUNPCKHQDQ Y3, Y3, Y3
|
|
VPXOR Y2, Y3, Y3
|
|
|
|
MOVQ e+16(FP), CX
|
|
VBROADCASTI128 (CX), Y2
|
|
|
|
VPXOR Y3, Y8, Y3
|
|
VPAND Y3, Y2, Y4
|
|
VPXOR Y4, Y0, Y8
|
|
VMOVDQA Y8, (BX)
|
|
VMOVDQA Y3, (AX)
|
|
RET
|