lib/rbtree_test.c: add (inorder) traversal test

This adds a second test for regular rb-tree testing in that there is no
need to repeat it for the augmented flavor.

Link: http://lkml.kernel.org/r/20170719014603.19029-6-dave@stgolabs.net
Signed-off-by: Davidlohr Bueso <dbueso@suse.de>
Signed-off-by: Andrew Morton <akpm@linux-foundation.org>
Signed-off-by: Linus Torvalds <torvalds@linux-foundation.org>
This commit is contained in:
Davidlohr Bueso 2017-09-08 16:14:49 -07:00 committed by Linus Torvalds
parent 223f8911ea
commit 977bd8d5e1

View File

@ -183,6 +183,7 @@ static int __init rbtree_test_init(void)
{ {
int i, j; int i, j;
cycles_t time1, time2, time; cycles_t time1, time2, time;
struct rb_node *node;
nodes = kmalloc(nnodes * sizeof(*nodes), GFP_KERNEL); nodes = kmalloc(nnodes * sizeof(*nodes), GFP_KERNEL);
if (!nodes) if (!nodes)
@ -206,8 +207,28 @@ static int __init rbtree_test_init(void)
time = time2 - time1; time = time2 - time1;
time = div_u64(time, perf_loops); time = div_u64(time, perf_loops);
printk(" -> %llu cycles\n", (unsigned long long)time); printk(" -> test 1 (latency of nnodes insert+delete): %llu cycles\n", (unsigned long long)time);
for (i = 0; i < nnodes; i++)
insert(nodes + i, &root);
time1 = get_cycles();
for (i = 0; i < perf_loops; i++) {
for (node = rb_first(&root); node; node = rb_next(node))
;
}
time2 = get_cycles();
time = time2 - time1;
time = div_u64(time, perf_loops);
printk(" -> test 2 (latency of inorder traversal): %llu cycles\n", (unsigned long long)time);
for (i = 0; i < nnodes; i++)
erase(nodes + i, &root);
/* run checks */
for (i = 0; i < check_loops; i++) { for (i = 0; i < check_loops; i++) {
init(); init();
for (j = 0; j < nnodes; j++) { for (j = 0; j < nnodes; j++) {
@ -238,7 +259,7 @@ static int __init rbtree_test_init(void)
time = time2 - time1; time = time2 - time1;
time = div_u64(time, perf_loops); time = div_u64(time, perf_loops);
printk(" -> %llu cycles\n", (unsigned long long)time); printk(" -> test 1 (latency of nnodes insert+delete): %llu cycles\n", (unsigned long long)time);
for (i = 0; i < check_loops; i++) { for (i = 0; i < check_loops; i++) {
init(); init();