From a12272c1ee778c7b0b143f24c9fc1d2690a54b9a Mon Sep 17 00:00:00 2001
From: Thomas Voss <mail@thomasvoss.com>
Date: Sat, 3 Aug 2024 23:20:29 +0200
Subject: Add isascii AVX512

---
 c/isascii-avx512/.gitignore |  1 +
 c/isascii-avx512/Makefile   |  8 ++++
 c/isascii-avx512/isascii.c  | 99 +++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 108 insertions(+)
 create mode 100644 c/isascii-avx512/.gitignore
 create mode 100644 c/isascii-avx512/Makefile
 create mode 100644 c/isascii-avx512/isascii.c

diff --git a/c/isascii-avx512/.gitignore b/c/isascii-avx512/.gitignore
new file mode 100644
index 0000000..4328e16
--- /dev/null
+++ b/c/isascii-avx512/.gitignore
@@ -0,0 +1 @@
+isascii
diff --git a/c/isascii-avx512/Makefile b/c/isascii-avx512/Makefile
new file mode 100644
index 0000000..3c49e43
--- /dev/null
+++ b/c/isascii-avx512/Makefile
@@ -0,0 +1,8 @@
+include ../base.mk
+
+all: isascii
+isascii: isascii.c
+	$(CC) $(CFLAGS) -lbsd -o $@ $<
+
+clean:
+	rm -f isascii
diff --git a/c/isascii-avx512/isascii.c b/c/isascii-avx512/isascii.c
new file mode 100644
index 0000000..612aa49
--- /dev/null
+++ b/c/isascii-avx512/isascii.c
@@ -0,0 +1,99 @@
+#include <sys/mman.h>
+#include <sys/stat.h>
+
+#include <err.h>
+#include <fcntl.h>
+#include <immintrin.h>
+#include <stdbool.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <time.h>
+#include <unistd.h>
+
+#include <bsd/stdlib.h>
+
+#define VECWDTH (512 / 8)
+
+static const unsigned char *readfile(const char *, size_t *);
+
+static const uint8_t charmsk[64] = {
+	0x80, 0x80, 0x80, 0x80, 0x80, 0x80, 0x80, 0x80,
+	0x80, 0x80, 0x80, 0x80, 0x80, 0x80, 0x80, 0x80,
+	0x80, 0x80, 0x80, 0x80, 0x80, 0x80, 0x80, 0x80,
+	0x80, 0x80, 0x80, 0x80, 0x80, 0x80, 0x80, 0x80,
+	0x80, 0x80, 0x80, 0x80, 0x80, 0x80, 0x80, 0x80,
+	0x80, 0x80, 0x80, 0x80, 0x80, 0x80, 0x80, 0x80,
+	0x80, 0x80, 0x80, 0x80, 0x80, 0x80, 0x80, 0x80,
+	0x80, 0x80, 0x80, 0x80, 0x80, 0x80, 0x80, 0x80,
+};
+
+bool
+strisascii(const unsigned char *s, size_t n)
+{
+	__m512i msk = _mm512_loadu_epi8(charmsk);
+	while (n >= VECWDTH) {
+		if (_mm512_test_epi8_mask(_mm512_loadu_epi8(s), msk) != 0)
+			return false;
+		s += VECWDTH;
+		n -= VECWDTH;
+	}
+	for (size_t i = 0; i < n; i++) {
+		if (s[i] > 0x7F)
+			return false;
+	}
+	return true;
+}
+
+bool
+strisascii_dumb(const unsigned char *s, size_t n)
+{
+	for (size_t i = 0; i < n; i++) {
+		if (s[i] > 0x7F)
+			return false;
+	}
+	return true;
+}
+
+int
+main(int argc, char **argv)
+{
+	if (argc != 2) {
+		fprintf(stderr, "Usage: %s file\n", getprogname());
+		exit(EXIT_FAILURE);
+	}
+
+	size_t len;
+	int rv = EXIT_SUCCESS;
+	const unsigned char *beg = readfile(argv[1], &len);
+
+	clock_t tmbeg = clock();
+	if (!strisascii(beg, len))
+		puts("Non-ASCII");
+	printf("Elapsed time: %.3fs\n", (double)(clock() - tmbeg) / CLOCKS_PER_SEC);
+
+	tmbeg = clock();
+	if (!strisascii_dumb((const unsigned char *)beg, len))
+		puts("Non-ASCII");
+	printf("Elapsed time: %.3fs\n", (double)(clock() - tmbeg) / CLOCKS_PER_SEC);
+
+	munmap((void *)beg, len);
+	return rv;
+}
+
+const unsigned char *
+readfile(const char *filename, size_t *n)
+{
+	int fd = open(filename, O_RDONLY);
+	if (fd == -1)
+		err(1, "open: %s", filename);
+
+	struct stat sb;
+	if (fstat(fd, &sb) == -1)
+		err(1, "fstat: %s", filename);
+
+	*n = sb.st_size;
+	const unsigned char *p = mmap(NULL, *n, PROT_READ, MAP_PRIVATE, fd, 0);
+	if (p == MAP_FAILED)
+		err(1, "mmap: %s", filename);
+	return p;
+}
-- 
cgit v1.2.3