103 	struct rb_root *root;  in page_root_per_function()  local
106 	root = xa_load(&dev->priv.page_root_xa, function);  in page_root_per_function()
107 	if (root)  in page_root_per_function()
108 		return root;  in page_root_per_function()
110 	root = kzalloc(sizeof(*root), GFP_KERNEL);  in page_root_per_function()
111 	if (!root)  in page_root_per_function()
114 	err = xa_insert(&dev->priv.page_root_xa, function, root, GFP_KERNEL);  in page_root_per_function()
116 		kfree(root);  in page_root_per_function()
120 	*root = RB_ROOT;  in page_root_per_function()
122 	return root;  in page_root_per_function()
128 	struct rb_root *root;  in insert_page()  local
134 	root = page_root_per_function(dev, function);  in insert_page()
135 	if (IS_ERR(root))  in insert_page()
136 		return PTR_ERR(root);  in insert_page()
138 	new = &root->rb_node;  in insert_page()
163 	rb_insert_color(&nfp->rb_node, root);  in insert_page()
173 	struct rb_root *root;  in find_fw_page()  local
177 	root = xa_load(&dev->priv.page_root_xa, function);  in find_fw_page()
178 	if (WARN_ON_ONCE(!root))  in find_fw_page()
181 	tmp = root->rb_node;  in find_fw_page()
257 	struct rb_root *root;  in free_fwp()  local
259 	root = xa_load(&dev->priv.page_root_xa, fwp->function);  in free_fwp()
260 	if (WARN_ON_ONCE(!root))  in free_fwp()
263 	rb_erase(&fwp->rb_node, root);  in free_fwp()
438 	struct rb_root *root;  in release_all_pages()  local
443 	root = xa_load(&dev->priv.page_root_xa, function);  in release_all_pages()
444 	if (WARN_ON_ONCE(!root))  in release_all_pages()
447 	p = rb_first(root);  in release_all_pages()
485 	struct rb_root *root;  in reclaim_pages_cmd()  local
501 	root = xa_load(&dev->priv.page_root_xa, get_function(func_id, ec_function));  in reclaim_pages_cmd()
502 	if (WARN_ON_ONCE(!root))  in reclaim_pages_cmd()
505 	p = rb_first(root);  in reclaim_pages_cmd()
684 				   struct rb_root *root, u32 function)  in mlx5_reclaim_root_pages()  argument
689 	while (!RB_EMPTY_ROOT(root)) {  in mlx5_reclaim_root_pages()
717 	struct rb_root *root;  in mlx5_reclaim_startup_pages()  local
722 		root = entry;  in mlx5_reclaim_startup_pages()
723 		mlx5_reclaim_root_pages(dev, root, id);  in mlx5_reclaim_startup_pages()
725 		kfree(root);  in mlx5_reclaim_startup_pages()