1 files changed, 8 insertions, 4 deletions

diff --git a/fs/btrfs/ctree.c b/fs/btrfs/ctree.c
index 77592931ab4f..ec7928a27aaa 100644
--- a/fs/btrfs/ctree.c
+++ b/fs/btrfs/ctree.c
@@ -19,6 +19,7 @@
 #include <linux/sched.h>
 #include <linux/slab.h>
 #include <linux/rbtree.h>
+#include <linux/vmalloc.h>
 #include "ctree.h"
 #include "disk-io.h"
 #include "transaction.h"
@@ -5361,10 +5362,13 @@ int btrfs_compare_trees(struct btrfs_root *left_root,
                 goto out;
         }
 
-        tmp_buf = kmalloc(left_root->nodesize, GFP_KERNEL);
+        tmp_buf = kmalloc(left_root->nodesize, GFP_KERNEL | __GFP_NOWARN);
         if (!tmp_buf) {
-                ret = -ENOMEM;
-                goto out;
+                tmp_buf = vmalloc(left_root->nodesize);
+                if (!tmp_buf) {
+                        ret = -ENOMEM;
+                        goto out;
+                }
         }
 
         left_path->search_commit_root = 1;
@@ -5565,7 +5569,7 @@ int btrfs_compare_trees(struct btrfs_root *left_root,
 out:
         btrfs_free_path(left_path);
         btrfs_free_path(right_path);
-        kfree(tmp_buf);
+        kvfree(tmp_buf);
         return ret;
 }