sched/headers: Prepare for new header dependencies before moving code to <linux/sched...

[mirror_ubuntu-artful-kernel.git] / mm / mmu_context.c

diff --git a/mm/mmu_context.c b/mm/mmu_context.c
index 6f4d27c5bb325f6468461b17cd8800cc1e473308..8888b124a386c5307fbb63a5380de4a8195e0919 100644 (file)
--- a/mm/mmu_context.c
+++ b/mm/mmu_context.c
@@ -5,6 +5,7 @@
 
 #include <linux/mm.h>
 #include <linux/sched.h>
+#include <linux/sched/mm.h>
 #include <linux/mmu_context.h>
 #include <linux/export.h>
 
@@ -25,7 +26,7 @@ void use_mm(struct mm_struct *mm)
        task_lock(tsk);
        active_mm = tsk->active_mm;
        if (active_mm != mm) {
-               atomic_inc(&mm->mm_count);
+               mmgrab(mm);
                tsk->active_mm = mm;
        }
        tsk->mm = mm;