--- /dev/null
+/*
+ * multiorder.c: Multi-order radix tree entry testing
+ * Copyright (c) 2016 Intel Corporation
+ * Author: Ross Zwisler <ross.zwisler@linux.intel.com>
+ * Author: Matthew Wilcox <matthew.r.wilcox@intel.com>
+ *
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms and conditions of the GNU General Public License,
+ * version 2, as published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
+ * more details.
+ */
+#include <linux/radix-tree.h>
+#include <linux/slab.h>
+#include <linux/errno.h>
+
+#include "test.h"
+
+static void multiorder_check(unsigned long index, int order)
+{
+       unsigned long i;
+       unsigned long min = index & ~((1UL << order) - 1);
+       unsigned long max = min + (1UL << order);
+       RADIX_TREE(tree, GFP_KERNEL);
+
+       printf("Multiorder index %ld, order %d\n", index, order);
+
+       assert(item_insert_order(&tree, index, order) == 0);
+
+       for (i = min; i < max; i++) {
+               struct item *item = item_lookup(&tree, i);
+               assert(item != 0);
+               assert(item->index == index);
+       }
+       for (i = 0; i < min; i++)
+               item_check_absent(&tree, i);
+       for (i = max; i < 2*max; i++)
+               item_check_absent(&tree, i);
+
+       assert(item_delete(&tree, index) != 0);
+
+       for (i = 0; i < 2*max; i++)
+               item_check_absent(&tree, i);
+}
+
+void multiorder_checks(void)
+{
+       int i;
+
+       for (i = 0; i < 20; i++) {
+               multiorder_check(200, i);
+               multiorder_check(0, i);
+               multiorder_check((1UL << i) + 1, i);
+       }
+}
 
        return radix_tree_tag_get(root, index, tag);
 }
 
-int __item_insert(struct radix_tree_root *root, struct item *item)
+int __item_insert(struct radix_tree_root *root, struct item *item,
+                       unsigned order)
 {
-       return radix_tree_insert(root, item->index, item);
+       return __radix_tree_insert(root, item->index, order, item);
 }
 
 int item_insert(struct radix_tree_root *root, unsigned long index)
 {
-       return __item_insert(root, item_create(index));
+       return __item_insert(root, item_create(index), 0);
+}
+
+int item_insert_order(struct radix_tree_root *root, unsigned long index,
+                       unsigned order)
+{
+       return __item_insert(root, item_create(index), order);
 }
 
 int item_delete(struct radix_tree_root *root, unsigned long index)
 
 };
 
 struct item *item_create(unsigned long index);
-int __item_insert(struct radix_tree_root *root, struct item *item);
+int __item_insert(struct radix_tree_root *root, struct item *item,
+                       unsigned order);
 int item_insert(struct radix_tree_root *root, unsigned long index);
+int item_insert_order(struct radix_tree_root *root, unsigned long index,
+                       unsigned order);
 int item_delete(struct radix_tree_root *root, unsigned long index);
 struct item *item_lookup(struct radix_tree_root *root, unsigned long index);
 
 void item_kill_tree(struct radix_tree_root *root);
 
 void tag_check(void);
+void multiorder_checks(void);
 
 struct item *
 item_tag_set(struct radix_tree_root *root, unsigned long index, int tag);