rtio: Add throughput test

Test throughput of submit and consume pair for rtio

Signed-off-by: Tom Burdick <[email protected]>

Author: teburd

tests/subsys/rtio/rtio_api/prj.conf

@@ -3,3 +3,4 @@ CONFIG_ZTEST_NEW_API=y
 CONFIG_LOG=y
 CONFIG_RTIO=y
 CONFIG_RTIO_SYS_MEM_BLOCKS=y
+CONFIG_TIMING_FUNCTIONS=y

tests/subsys/rtio/rtio_api/src/test_rtio_api.c

@@ -13,13 +13,13 @@
 #include <zephyr/app_memory/mem_domain.h>
 #include <zephyr/sys/util_loops.h>
 #include <zephyr/sys/time_units.h>
-#include <zephyr/rtio/rtio_mpsc.h>
+#include <zephyr/timing/timing.h>
 #include <zephyr/rtio/rtio.h>
 
 #include "rtio_iodev_test.h"
 
 /* Repeat tests to ensure they are repeatable */
-#define TEST_REPEATS 2
+#define TEST_REPEATS 4
 
 #define MEM_BLK_COUNT 4
 #define MEM_BLK_SIZE 16
@@ -182,7 +182,8 @@ void test_rtio_multiple_chains_(struct rtio *r)
 			cqe = rtio_cqe_consume(r);
 		}
 
-		TC_PRINT("consumed cqe %p, result, %d, userdata %lu\n", cqe, cqe-> result, (uintptr_t)cqe->userdata);
+		TC_PRINT("consumed cqe %p, result, %d, userdata %lu\n", cqe,
+			 cqe->result, (uintptr_t)cqe->userdata);
 
 		zassert_not_null(cqe, "Expected a valid cqe");
 		zassert_ok(cqe->result, "Result should be ok");
@@ -427,6 +428,44 @@ ZTEST(rtio_api, test_rtio_transaction)
 	}
 }
 
+#define THROUGHPUT_ITERS 100000
+RTIO_DEFINE(r_throughput, 4, 4);
+
+void _test_rtio_throughput(struct rtio *r)
+{
+	timing_t start_time, end_time;
+	struct rtio_cqe *cqe;
+	struct rtio_sqe *sqe;
+
+	timing_init();
+	timing_start();
+
+	start_time = timing_counter_get();
+
+	for (uint32_t i = 0; i < THROUGHPUT_ITERS; i++) {
+		sqe = rtio_sqe_acquire(r);
+		rtio_sqe_prep_nop(sqe, NULL, NULL);
+		rtio_submit(r, 0);
+		cqe = rtio_cqe_consume(r);
+		rtio_cqe_release(r, cqe);
+	}
+
+	end_time = timing_counter_get();
+
+	uint64_t cycles = timing_cycles_get(&start_time, &end_time);
+	uint64_t ns = timing_cycles_to_ns(cycles);
+
+	TC_PRINT("%llu ns for %d iterations, %llu ns per op\n",
+		 ns, THROUGHPUT_ITERS, ns/THROUGHPUT_ITERS);
+}
+
+
+ZTEST(rtio_api, test_rtio_throughput)
+{
+	_test_rtio_throughput(&r_throughput);
+}
+
+
 static void *rtio_api_setup(void)
 {
 #ifdef CONFIG_USERSPACE

tests/subsys/rtio/rtio_api/src/test_rtio_mpsc.c

@@ -7,6 +7,7 @@
 #include <zephyr/ztest.h>
 #include <zephyr/kernel.h>
 #include <zephyr/sys/util_loops.h>
+#include <zephyr/timing/timing.h>
 #include <zephyr/rtio/rtio_spsc.h>
 #include <zephyr/rtio/rtio_mpsc.h>
 
@@ -182,4 +183,32 @@ ZTEST(rtio_mpsc, test_mpsc_threaded)
 	}
 }
 
+#define THROUGHPUT_ITERS 100000
+
+ZTEST(rtio_mpsc, test_mpsc_throughput)
+{
+	struct rtio_mpsc_node node;
+	timing_t start_time, end_time;
+
+	rtio_mpsc_init(&mpsc_q);
+	timing_init();
+	timing_start();
+
+	start_time = timing_counter_get();
+
+	for (int i = 0; i < THROUGHPUT_ITERS; i++) {
+		rtio_mpsc_push(&mpsc_q, &node);
+
+		rtio_mpsc_pop(&mpsc_q);
+	}
+
+	end_time = timing_counter_get();
+
+	uint64_t cycles = timing_cycles_get(&start_time, &end_time);
+	uint64_t ns = timing_cycles_to_ns(cycles);
+
+	TC_PRINT("%llu ns for %d iterations, %llu ns per op\n", ns,
+		 THROUGHPUT_ITERS, ns/THROUGHPUT_ITERS);
+}
+
 ZTEST_SUITE(rtio_mpsc, NULL, NULL, NULL, NULL, NULL);

tests/subsys/rtio/rtio_api/src/test_rtio_spsc.c

@@ -5,6 +5,7 @@
  */
 
 #include <zephyr/ztest.h>
+#include <zephyr/timing/timing.h>
 #include <zephyr/rtio/rtio_spsc.h>
 #include "rtio_api.h"
 
@@ -215,4 +216,36 @@ ZTEST(rtio_spsc, test_spsc_threaded)
 	k_thread_join(tinfo[0].tid, K_FOREVER);
 }
 
+#define THROUGHPUT_ITERS 100000
+
+ZTEST(rtio_spsc, test_spsc_throughput)
+{
+	timing_t start_time, end_time;
+
+	timing_init();
+	timing_start();
+
+	start_time = timing_counter_get();
+
+	uint32_t *x, *y;
+
+	for (int i = 0; i < THROUGHPUT_ITERS; i++) {
+		x = rtio_spsc_acquire(&spsc);
+		*x = i;
+		rtio_spsc_produce(&spsc);
+
+		y = rtio_spsc_consume(&spsc);
+		rtio_spsc_release(&spsc);
+	}
+
+	end_time = timing_counter_get();
+
+	uint64_t cycles = timing_cycles_get(&start_time, &end_time);
+	uint64_t ns = timing_cycles_to_ns(cycles);
+
+	TC_PRINT("%llu ns for %d iterations, %llu ns per op\n", ns,
+		 THROUGHPUT_ITERS, ns/THROUGHPUT_ITERS);
+}
+
+
 ZTEST_SUITE(rtio_spsc, NULL, NULL, NULL, NULL, NULL);