* |- Push harder to find free extents
  *    |- If not found, re-iterate all block groups
  */
-static noinline int find_free_extent(struct btrfs_fs_info *fs_info,
+static noinline int find_free_extent(struct btrfs_root *root,
                                u64 ram_bytes, u64 num_bytes, u64 empty_size,
                                u64 hint_byte, struct btrfs_key *ins,
                                u64 flags, int delalloc)
 {
+       struct btrfs_fs_info *fs_info = root->fs_info;
        int ret = 0;
        int cache_block_group_error = 0;
        struct btrfs_free_cluster *last_ptr = NULL;
        ins->objectid = 0;
        ins->offset = 0;
 
-       trace_find_free_extent(fs_info, num_bytes, empty_size, flags);
+       trace_find_free_extent(root, num_bytes, empty_size, flags);
 
        space_info = btrfs_find_space_info(fs_info, flags);
        if (!space_info) {
        flags = get_alloc_profile_by_root(root, is_data);
 again:
        WARN_ON(num_bytes < fs_info->sectorsize);
-       ret = find_free_extent(fs_info, ram_bytes, num_bytes, empty_size,
+       ret = find_free_extent(root, ram_bytes, num_bytes, empty_size,
                               hint_byte, ins, flags, delalloc);
        if (!ret && !is_data) {
                btrfs_dec_block_group_reservations(fs_info, ins->objectid);
 
 
 TRACE_EVENT(find_free_extent,
 
-       TP_PROTO(const struct btrfs_fs_info *fs_info, u64 num_bytes,
+       TP_PROTO(const struct btrfs_root *root, u64 num_bytes,
                 u64 empty_size, u64 data),
 
-       TP_ARGS(fs_info, num_bytes, empty_size, data),
+       TP_ARGS(root, num_bytes, empty_size, data),
 
        TP_STRUCT__entry_btrfs(
+               __field(        u64,    root_objectid           )
                __field(        u64,    num_bytes               )
                __field(        u64,    empty_size              )
                __field(        u64,    data                    )
        ),
 
-       TP_fast_assign_btrfs(fs_info,
+       TP_fast_assign_btrfs(root->fs_info,
+               __entry->root_objectid  = root->root_key.objectid;
                __entry->num_bytes      = num_bytes;
                __entry->empty_size     = empty_size;
                __entry->data           = data;
        ),
 
        TP_printk_btrfs("root=%llu(%s) len=%llu empty_size=%llu flags=%llu(%s)",
-                 show_root_type(BTRFS_EXTENT_TREE_OBJECTID),
+                 show_root_type(__entry->root_objectid),
                  __entry->num_bytes, __entry->empty_size, __entry->data,
                  __print_flags((unsigned long)__entry->data, "|",
                                 BTRFS_GROUP_FLAGS))