1 files changed, 12 insertions, 6 deletions

diff --git a/mm/vmalloc.c b/mm/vmalloc.c
index e5e9e1fcac01..072c8e0df90a 100644
--- a/mm/vmalloc.c
+++ b/mm/vmalloc.c
@@ -662,23 +662,27 @@ static bool __purge_vmap_area_lazy(unsigned long start, unsigned long end)
         struct llist_node *valist;
         struct vmap_area *va;
         struct vmap_area *n_va;
-        bool do_free = false;
+        int resched_threshold;
 
         lockdep_assert_held(&vmap_purge_lock);
 
         valist = llist_del_all(&vmap_purge_list);
+        if (unlikely(valist == NULL))
+                return false;
+
+        /*
+         * TODO: to calculate a flush range without looping.
+         * The list can be up to lazy_max_pages() elements.
+         */
         llist_for_each_entry(va, valist, purge_list) {
                 if (va->va_start < start)
                         start = va->va_start;
                 if (va->va_end > end)
                         end = va->va_end;
-                do_free = true;
         }
 
-        if (!do_free)
-                return false;
-
         flush_tlb_kernel_range(start, end);
+        resched_threshold = (int) lazy_max_pages() << 1;
 
         spin_lock(&vmap_area_lock);
         llist_for_each_entry_safe(va, n_va, valist, purge_list) {
@@ -686,7 +690,9 @@ static bool __purge_vmap_area_lazy(unsigned long start, unsigned long end)
 
                 __free_vmap_area(va);
                 atomic_sub(nr, &vmap_lazy_nr);
-                cond_resched_lock(&vmap_area_lock);
+
+                if (atomic_read(&vmap_lazy_nr) < resched_threshold)
+                        cond_resched_lock(&vmap_area_lock);
         }
         spin_unlock(&vmap_area_lock);
         return true;