rtio: Add throughput test (fc32f1c0) · Commits · Wenxi XU / zephyr

tests/subsys/rtio/rtio_api/prj.conf

+1 −0

Original line number	Diff line number	Diff line
		@@ -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

+42 −3

Original line number	Diff line number	Diff line
		@@ -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

+29 −0

Original line number	Diff line number	Diff line
		@@ -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

+33 −0

Original line number	Diff line number	Diff line
		@@ -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);

Admin message