forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-07-23 05:06:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

bitutil: optimize fastORBytes with bit shifts

Browse source
This commit is contained in:
Sahil-4555 2025-08-28 15:54:08 +05:30
parent e77dc6511f
commit ec64990c98
1 changed files with 20 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
common/bitutil/bitutil.go
View file

@ -134,21 +134,32 @@ func ORBytes(dst, a, b []byte) int {
// fastORBytes ors in bulk. It only works on architectures that support // fastORBytes ors in bulk. It only works on architectures that support
// unaligned read/writes. // unaligned read/writes.
func fastORBytes(dst, a, b []byte) int { func fastORBytes(dst, a, b []byte) int {
n := len(a) n := min(len(b), len(a))
if len(b) < n {
n = len(b)
}
w := n / wordSize w := n / wordSize
if w > 0 { if w > 0 {
dw := *(*[]uintptr)(unsafe.Pointer(&dst)) dw := *(*[]uintptr)(unsafe.Pointer(&dst))
aw := *(*[]uintptr)(unsafe.Pointer(&a)) aw := *(*[]uintptr)(unsafe.Pointer(&a))
bw := *(*[]uintptr)(unsafe.Pointer(&b)) bw := *(*[]uintptr)(unsafe.Pointer(&b))
for i := 0; i < w; i++ {
i := 0
for i <= w-4 {
dw[i] = aw[i] | bw[i] dw[i] = aw[i] | bw[i]
dw[i+1] = aw[i+1] | bw[i+1]
dw[i+2] = aw[i+2] | bw[i+2]
dw[i+3] = aw[i+3] | bw[i+3]
i += 4
}
// Handle remaining words
for i < w {
dw[i] = aw[i] | bw[i]
i++
} }
} }
for i := n - n%wordSize; i < n; i++ {
dst[i] = a[i] | b[i] // Handle remaining bytes
start := w * wordSize
for i := 0; i < n&(wordSize-1); i++ {
dst[start+i] = a[start+i] | b[start+i]
} }
return n return n
} }
@ -156,10 +167,7 @@ func fastORBytes(dst, a, b []byte) int {
// safeORBytes ors one by one. It works on all architectures, independent if // safeORBytes ors one by one. It works on all architectures, independent if
// it supports unaligned read/writes or not. // it supports unaligned read/writes or not.
func safeORBytes(dst, a, b []byte) int { func safeORBytes(dst, a, b []byte) int {
n := len(a) n := min(len(b), len(a))
if len(b) < n {
n = len(b)
}
for i := 0; i < n; i++ { for i := 0; i < n; i++ {
dst[i] = a[i] | b[i] dst[i] = a[i] | b[i]
} }
@ -198,7 +206,7 @@ func fastTestBytes(p []byte) bool {
// safeTestBytes tests for set bits one byte at a time. It works on all // safeTestBytes tests for set bits one byte at a time. It works on all
// architectures, independent if it supports unaligned read/writes or not. // architectures, independent if it supports unaligned read/writes or not.
func safeTestBytes(p []byte) bool { func safeTestBytes(p []byte) bool {
for i := 0; i < len(p); i++ { for i := range p {
if p[i] != 0 { if p[i] != 0 {
return true return true
} }