61 lines
1.5 KiB
ArmAsm
61 lines
1.5 KiB
ArmAsm
#include "textflag.h"
|
|
|
|
#define mask(bit, tmp, to) \
|
|
VPSRLW bit, Y10, tmp \
|
|
VPAND Y12, tmp, to \ // to = 0x000<bit>000<bit>...
|
|
VPSUBW to, Y13, to // to = 0xFFFF.. or 0x0000 depending on bit
|
|
|
|
#define mulBit(bit, in_1, in_2, out_1, out_2) \
|
|
VPSLLQ $1, in_1, Y1 \
|
|
VPALIGNR $8, Y1, in_1, Y2 \
|
|
VPSRLQ $63, Y2, Y2 \
|
|
VPXOR Y1, Y2, Y2 \
|
|
VPAND Y1, Y14, Y3 \
|
|
VPUNPCKHQDQ Y3, Y3, Y3 \
|
|
VPXOR Y2, Y3, Y3 \
|
|
mask(bit, Y11, Y2) \
|
|
VPXOR Y3, in_2, out_1 \
|
|
VPAND out_1, Y2, Y4 \
|
|
VPXOR Y4, in_1, out_2 \
|
|
|
|
// func mulByteSliceRightx2(c00c10, c01c11 *[4]uint64, n int, data *byte)
|
|
TEXT ·mulByteSliceRightx2(SB), NOSPLIT, $0
|
|
MOVQ c00c10+0(FP), AX
|
|
VMOVDQU (AX), Y0
|
|
MOVQ c01c11+8(FP), BX
|
|
VMOVDQU (BX), Y8
|
|
|
|
VPXOR Y13, Y13, Y13 // Y13 = 0x0000...
|
|
VPCMPEQB Y14, Y14, Y14 // Y14 = 0xFFFF...
|
|
VPSUBQ Y14, Y13, Y10
|
|
VPSUBW Y14, Y13, Y12 // Y12 = 0x00010001... (packed words of 1)
|
|
VPSLLQ $63, Y10, Y14 // Y14 = 0x10000000... (packed quad-words with HSB set)
|
|
|
|
MOVQ n+16(FP), CX
|
|
MOVQ data+24(FP), DX
|
|
|
|
loop:
|
|
CMPQ CX, $0
|
|
JEQ finish
|
|
SUBQ $1, CX
|
|
|
|
VPBROADCASTB (DX), X10 // X10 = packed bytes of b.
|
|
VPMOVZXBW X10, Y10 // Extend with zeroes to packed words.
|
|
ADDQ $1, DX
|
|
|
|
mulBit($7, Y0, Y8, Y5, Y6)
|
|
mulBit($6, Y5, Y6, Y0, Y8)
|
|
mulBit($5, Y0, Y8, Y5, Y6)
|
|
mulBit($4, Y5, Y6, Y0, Y8)
|
|
mulBit($3, Y0, Y8, Y5, Y6)
|
|
mulBit($2, Y5, Y6, Y0, Y8)
|
|
mulBit($1, Y0, Y8, Y5, Y6)
|
|
mulBit($0, Y5, Y6, Y0, Y8)
|
|
|
|
JMP loop
|
|
|
|
finish:
|
|
VMOVDQU Y8, (BX)
|
|
VMOVDQU Y0, (AX)
|
|
|
|
RET
|