mirror of
https://github.com/featurebasedb/featurebase.git
synced 2026-09-05 16:15:56 +00:00
08659c350b
This commit moves the computation of the intersection count to the `roaring` package so that no allocations are required. The implementation operates at the roaring container level and has specialized functions for array-array, array-bitmap, and bitmap-bitmap container pairs.
122 lines
2 KiB
ArmAsm
122 lines
2 KiB
ArmAsm
#include "textflag.h"
|
|
|
|
TEXT ·hasAsm(SB),4,$0
|
|
MOVQ $1, AX
|
|
CPUID
|
|
SHRQ $23, CX
|
|
ANDQ $1, CX
|
|
MOVB CX, ret+0(FP)
|
|
RET
|
|
|
|
TEXT ·POPCNTQ(SB),NOSPLIT,$0-8
|
|
MOVQ x+0(FP), BP
|
|
POPCNTQ BP, BX
|
|
MOVQ BX, ret+8(FP)
|
|
RET
|
|
|
|
TEXT ·BSFQ(SB),NOSPLIT,$0-8
|
|
MOVQ x+0(FP), BP
|
|
BSFQ BP, BX
|
|
MOVQ BX, ret+8(FP)
|
|
RET
|
|
|
|
#define POPCNTQ_DX_DX BYTE $0xf3; BYTE $0x48; BYTE $0x0f; BYTE $0xb8; BYTE $0xd2
|
|
|
|
TEXT ·popcntSliceAsm(SB),4,$0-32
|
|
XORQ AX, AX
|
|
MOVQ s+0(FP), SI
|
|
MOVQ s+8(FP), CX
|
|
TESTQ CX, CX
|
|
JZ popcntSliceEnd
|
|
popcntSliceLoop:
|
|
BYTE $0xf3; BYTE $0x48; BYTE $0x0f; BYTE $0xb8; BYTE $0x16 // POPCNTQ (SI), DX
|
|
ADDQ DX, AX
|
|
ADDQ $8, SI
|
|
LOOP popcntSliceLoop
|
|
popcntSliceEnd:
|
|
MOVQ AX, ret+24(FP)
|
|
RET
|
|
|
|
TEXT ·popcntMaskSliceAsm(SB),4,$0-56
|
|
XORQ AX, AX
|
|
MOVQ s+0(FP), SI
|
|
MOVQ s+8(FP), CX
|
|
TESTQ CX, CX
|
|
JZ popcntMaskSliceEnd
|
|
MOVQ m+24(FP), DI
|
|
popcntMaskSliceLoop:
|
|
MOVQ (DI), DX
|
|
NOTQ DX
|
|
ANDQ (SI), DX
|
|
POPCNTQ_DX_DX
|
|
ADDQ DX, AX
|
|
ADDQ $8, SI
|
|
ADDQ $8, DI
|
|
LOOP popcntMaskSliceLoop
|
|
popcntMaskSliceEnd:
|
|
MOVQ AX, ret+48(FP)
|
|
RET
|
|
|
|
TEXT ·popcntAndSliceAsm(SB),4,$0-56
|
|
XORQ AX, AX
|
|
MOVQ s+0(FP), SI
|
|
MOVQ s+8(FP), CX
|
|
TESTQ CX, CX
|
|
JZ popcntAndSliceEnd
|
|
MOVQ m+24(FP), DI
|
|
popcntAndSliceLoop:
|
|
MOVQ (DI), DX
|
|
ANDQ (SI), DX
|
|
POPCNTQ_DX_DX
|
|
ADDQ DX, AX
|
|
ADDQ $8, SI
|
|
ADDQ $8, DI
|
|
LOOP popcntAndSliceLoop
|
|
popcntAndSliceEnd:
|
|
MOVQ AX, ret+48(FP)
|
|
RET
|
|
|
|
TEXT ·popcntOrSliceAsm(SB),4,$0-56
|
|
XORQ AX, AX
|
|
MOVQ s+0(FP), SI
|
|
MOVQ s+8(FP), CX
|
|
TESTQ CX, CX
|
|
JZ popcntOrSliceEnd
|
|
MOVQ m+24(FP), DI
|
|
popcntOrSliceLoop:
|
|
MOVQ (DI), DX
|
|
ORQ (SI), DX
|
|
POPCNTQ_DX_DX
|
|
ADDQ DX, AX
|
|
ADDQ $8, SI
|
|
ADDQ $8, DI
|
|
LOOP popcntOrSliceLoop
|
|
popcntOrSliceEnd:
|
|
MOVQ AX, ret+48(FP)
|
|
RET
|
|
|
|
TEXT ·popcntXorSliceAsm(SB),4,$0-56
|
|
XORQ AX, AX
|
|
MOVQ s+0(FP), SI
|
|
MOVQ s+8(FP), CX
|
|
TESTQ CX, CX
|
|
JZ popcntXorSliceEnd
|
|
MOVQ m+24(FP), DI
|
|
popcntXorSliceLoop:
|
|
MOVQ (DI), DX
|
|
XORQ (SI), DX
|
|
POPCNTQ_DX_DX
|
|
ADDQ DX, AX
|
|
ADDQ $8, SI
|
|
ADDQ $8, DI
|
|
LOOP popcntXorSliceLoop
|
|
popcntXorSliceEnd:
|
|
MOVQ AX, ret+48(FP)
|
|
RET
|
|
|
|
|
|
TEXT ·popcntAsm(SB),4,$0-16
|
|
MOVQ x+0(FP), DX
|
|
POPCNTQ_DX_DX
|
|
MOVQ DX, ret+8(FP)
|
|
RET
|