s390: gmap, fix rcu lock

Signed-off-by: Liam R. Howlett <Liam.Howlett@Oracle.com>

diff --git a/arch/s390/mm/gmap.c b/arch/s390/mm/gmap.c
index d8f3f4d409652c889ba8b0688fa86f86d8427ffa..77879744d65236f678e473871722c32ac50877ac 100644 (file)
--- a/arch/s390/mm/gmap.c
+++ b/arch/s390/mm/gmap.c
@@ -2504,6 +2504,7 @@ static inline void thp_split_mm(struct mm_struct *mm)
        struct vm_area_struct *vma;
        MA_STATE(mas, &mm->mm_mt, 0, 0);
 
+       /* Must hold mm_mt lock already */
        mas_for_each(&mas, vma, ULONG_MAX) {
                vma->vm_flags &= ~VM_HUGEPAGE;
                vma->vm_flags |= VM_NOHUGEPAGE;
@@ -2574,6 +2575,7 @@ int gmap_mark_unmergeable(void)
        int ret;
        MA_STATE(mas, &mm->mm_mt, 0, 0);
 
+       /* Must hold mm_mt lock already */
        mas_for_each(&mas, vma, ULONG_MAX) {
                ret = ksm_madvise(vma, vma->vm_start, vma->vm_end,
                                  MADV_UNMERGEABLE, &vma->vm_flags);