From 01c6e51e6aaaf1e10aade8e5215a77ad09482efe Mon Sep 17 00:00:00 2001
From: Rokas Puzonas <rokas.puz@gmail.com>
Date: Thu, 15 Feb 2024 21:03:11 +0200
Subject: [PATCH] add load uop tests

---
 Makefile              |  7 ++++--
 src/load_uop.asm      | 56 +++++++++++++++++++++++++++++++++++++++++++
 src/load_uop.h        |  6 +++++
 src/main.c            |  3 +++
 src/main_load_uop.c   | 40 +++++++++++++++++++++++++++++++
 src/main_write_loop.c |  2 --
 6 files changed, 110 insertions(+), 4 deletions(-)
 create mode 100644 src/load_uop.asm
 create mode 100644 src/load_uop.h
 create mode 100644 src/main_load_uop.c

diff --git a/Makefile b/Makefile
index 6694c52..35c0975 100644
--- a/Makefile
+++ b/Makefile
@@ -1,8 +1,11 @@
 CFLAGS=-lm -g -Wall -O1
 
-build/main: src/main.c src/repetition_tester.c build/multi_nop_loop.o build/write_loops.o src/rprof.h
+build/main: src/main.c src/repetition_tester.c build/multi_nop_loop.o build/write_loops.o build/load_uop.o src/rprof.h
 	mkdir -p build
-	gcc -o build/main src/main.c build/multi_nop_loop.o build/write_loops.o $(CFLAGS)
+	gcc -o build/main src/main.c build/multi_nop_loop.o build/write_loops.o build/load_uop.o $(CFLAGS)
+
+build/load_uop.o: src/load_uop.asm
+	nasm -g -f elf64 -o build/load_uop.o src/load_uop.asm
 
 build/write_loops.o: src/write_loops.asm
 	nasm -g -f elf64 -o build/write_loops.o src/write_loops.asm
diff --git a/src/load_uop.asm b/src/load_uop.asm
new file mode 100644
index 0000000..03f9cf9
--- /dev/null
+++ b/src/load_uop.asm
@@ -0,0 +1,56 @@
+global mov_load_x1
+global mov_load_x2
+global mov_load_x3
+global mov_load_x4
+
+section .text
+
+; rsi - byte_count
+; rdi - buffer
+mov_load_x1:
+	mov rcx, rsi
+	align 64
+.loop:
+	mov rax, [rdi]
+	sub rcx, 1
+	jnle .loop
+	ret
+
+; rsi - byte_count
+; rdi - buffer
+mov_load_x2:
+	mov rcx, rsi
+	align 64
+.loop:
+	mov rax, [rdi]
+	mov rax, [rdi]
+	sub rcx, 2
+	jnle .loop
+	ret
+
+; rsi - byte_count
+; rdi - buffer
+mov_load_x3:
+	mov rcx, rsi
+	align 64
+.loop:
+	mov rax, [rdi]
+	mov rax, [rdi]
+	mov rax, [rdi]
+	sub rcx, 3
+	jnle .loop
+	ret
+
+; rsi - byte_count
+; rdi - buffer
+mov_load_x4:
+	mov rcx, rsi
+	align 64
+.loop:
+	mov rax, [rdi]
+	mov rax, [rdi]
+	mov rax, [rdi]
+	mov rax, [rdi]
+	sub rcx, 4
+	jnle .loop
+	ret
diff --git a/src/load_uop.h b/src/load_uop.h
new file mode 100644
index 0000000..d17d8c3
--- /dev/null
+++ b/src/load_uop.h
@@ -0,0 +1,6 @@
+#include <stdint.h>
+
+void mov_load_x1(uint8_t *buffer, uint64_t byte_count);
+void mov_load_x2(uint8_t *buffer, uint64_t byte_count);
+void mov_load_x3(uint8_t *buffer, uint64_t byte_count);
+void mov_load_x4(uint8_t *buffer, uint64_t byte_count);
diff --git a/src/main.c b/src/main.c
index dc1063e..1697acc 100644
--- a/src/main.c
+++ b/src/main.c
@@ -14,6 +14,7 @@
 #include "main_malloc_read.c"
 #include "main_write_backward.c"
 #include "main_write_loop.c"
+#include "main_load_uop.c"
 
 int main(int argc, char **argv) {
 	if (argc < 2) {
@@ -27,6 +28,8 @@ int main(int argc, char **argv) {
 		return main_test_write_bytes_asm();
 	} else if (!strncmp(test_name, "write_all_bytes", sizeof("write_bytes"))) {
 		return main_test_write_all_bytes();
+	} else if (!strncmp(test_name, "load_uop", sizeof("load_uop"))) {
+		return main_test_load_uop();
 	} else if (!strncmp(test_name, "write_loop", sizeof("write_loop"))) {
 		return main_test_write_loop();
 	} else if (!strncmp(test_name, "write_backward", sizeof("write_backward"))) {
diff --git a/src/main_load_uop.c b/src/main_load_uop.c
new file mode 100644
index 0000000..84b7a49
--- /dev/null
+++ b/src/main_load_uop.c
@@ -0,0 +1,40 @@
+#include "repetition_tester.c"
+#include "load_uop.h"
+
+int main_test_load_uop() {
+	typedef void (*test_cb)(uint8_t *buffer, uint64_t byte_count);
+	struct testcase {
+		char *name;
+		test_cb cb;
+	};
+
+	struct testcase cases[] = {
+		{ .name = "mov_load_x1()", .cb = mov_load_x1 },
+		{ .name = "mov_load_x2()", .cb = mov_load_x2 },
+		{ .name = "mov_load_x3()", .cb = mov_load_x3 },
+		{ .name = "mov_load_x4()", .cb = mov_load_x4 },
+	};
+
+	struct repetitor repetitor = {};
+	repetitor_init(&repetitor);
+	printf("CPU Frequency: %ldHz (~%.2fGHz)\n", repetitor.cpu_freq, (float)repetitor.cpu_freq/(1000*1000*1000));
+
+	uint64_t byte_count = 4096 * 1024;
+	uint8_t buffer[byte_count];
+
+	for (int i = 0; i < ARRAY_LEN(cases); i++) {
+		struct testcase *testcase = &cases[i];
+		repetitor_clear(&repetitor);
+		while (repetitor_repeat(&repetitor, 2)) {
+			repetitor_start(&repetitor);
+			repetitor_measure_start(&repetitor);
+			testcase->cb(buffer, byte_count);
+			repetitor_measure_stop(&repetitor, byte_count);
+			repetitor_stop(&repetitor);
+		}
+		repetitor_print_results_label(&repetitor, testcase->name);
+	}
+
+	return 0;
+}
+
diff --git a/src/main_write_loop.c b/src/main_write_loop.c
index e3d17b8..6d65bcd 100644
--- a/src/main_write_loop.c
+++ b/src/main_write_loop.c
@@ -53,6 +53,4 @@ int main_test_write_loop() {
 	}
 
 	return 0;
-
-	return 0;
 }