diff --git a/include/linux/mman.h b/include/linux/mman.h
index d35d984c058cafc44c44655350fe75da826422fd..a8ea591faed7a6da98139a2f2d0fc0dbee9af218 100644
--- a/include/linux/mman.h
+++ b/include/linux/mman.h
@@ -76,15 +76,16 @@ static inline int dvpp_mmap_zone(unsigned long addr) { return 0; }
 #ifdef CONFIG_COHERENT_DEVICE
 #define CHECKNODE_BITS  48
 #define CHECKNODE_MASK	(~((_AC(1, UL) << CHECKNODE_BITS) - 1))
-static inline void set_vm_checknode(vm_flags_t vm_flags, unsigned long flags)
+static inline void set_vm_checknode(vm_flags_t *vm_flags, unsigned long flags)
 {
 	if (is_set_cdmmask())
-		vm_flags |= VM_CHECKNODE | ((((flags >> MAP_HUGE_SHIFT) &
+		*vm_flags |= VM_CHECKNODE | ((((flags >> MAP_HUGE_SHIFT) &
 			MAP_HUGE_MASK) << CHECKNODE_BITS) & CHECKNODE_MASK);
 }
 #else
 #define CHECKNODE_BITS	(0)
-static inline void set_vm_checknode(vm_flags_t vm_flags, unsigned long flags) {}
+static inline void set_vm_checknode(vm_flags_t *vm_flags, unsigned long flags)
+{}
 #endif
 
 /*
diff --git a/mm/mmap.c b/mm/mmap.c
index 4cc9ee8a02873fa1028ea1c3d2baff519e820c8b..04e34c0227755d36fff7740d1200e78c281e7e8f 100644
--- a/mm/mmap.c
+++ b/mm/mmap.c
@@ -1565,7 +1565,7 @@ unsigned long do_mmap(struct file *file, unsigned long addr,
 	 * hugetlbfs file mmap will use it to check node
 	 */
 	if (flags & MAP_CHECKNODE)
-		set_vm_checknode(vm_flags, flags);
+		set_vm_checknode(&vm_flags, flags);
 
 	addr = mmap_region(file, addr, len, vm_flags, pgoff, uf);
 	if (!IS_ERR_VALUE(addr) &&