Merge pull request #20 from Jille/avx1

Add AVX (not AVX2) implementations
bwesterb · Sep 2, 2024 · 344ccb8 · 344ccb8
2 parents a6c99a1 + 731b398
commit 344ccb8
Show file tree

Hide file tree

Showing 7 changed files with 476 additions and 138 deletions.
diff --git a/and_amd64.go b/and_amd64.go
@@ -10,6 +10,12 @@ func and(dst, a, b []byte) {
 			andAVX2(&dst[0], &a[0], &b[0], l)
 		}
 		l <<= 8
+	} else if hasAVX() {
+		l = uint64(len(a)) >> 7
+		if l != 0 {
+			andAVX(&dst[0], &a[0], &b[0], l)
+		}
+		l <<= 7
 	}
 	andGeneric(dst[l:], a[l:], b[l:])
 }
@@ -22,6 +28,12 @@ func or(dst, a, b []byte) {
 			orAVX2(&dst[0], &a[0], &b[0], l)
 		}
 		l <<= 8
+	} else if hasAVX() {
+		l = uint64(len(a)) >> 7
+		if l != 0 {
+			orAVX(&dst[0], &a[0], &b[0], l)
+		}
+		l <<= 7
 	}
 	orGeneric(dst[l:], a[l:], b[l:])
 }
@@ -34,6 +46,12 @@ func xor(dst, a, b []byte) {
 			xorAVX2(&dst[0], &a[0], &b[0], l)
 		}
 		l <<= 8
+	} else if hasAVX() {
+		l = uint64(len(a)) >> 7
+		if l != 0 {
+			xorAVX(&dst[0], &a[0], &b[0], l)
+		}
+		l <<= 7
 	}
 	xorGeneric(dst[l:], a[l:], b[l:])
 }
@@ -46,6 +64,12 @@ func andNot(dst, a, b []byte) {
 			andNotAVX2(&dst[0], &a[0], &b[0], l)
 		}
 		l <<= 8
+	} else if hasAVX() {
+		l = uint64(len(a)) >> 7
+		if l != 0 {
+			andNotAVX(&dst[0], &a[0], &b[0], l)
+		}
+		l <<= 7
 	}
 	andNotGeneric(dst[l:], a[l:], b[l:])
 }
@@ -72,6 +96,12 @@ func memset(dst []byte, b byte) {
 			memsetAVX2(&dst[0], l, b)
 		}
 		l <<= 5
+	} else if hasAVX() {
+		l = uint64(len(dst)) >> 4
+		if l != 0 {
+			memsetAVX(&dst[0], l, b)
+		}
+		l <<= 4
 	}
 	memsetGeneric(dst[l:], b)
 }