1 files changed, 17 insertions, 3 deletions

diff --git a/include/asm-s390/tlbflush.h b/include/asm-s390/tlbflush.h
index a5e876539ec6..70fa5ae58180 100644
--- a/include/asm-s390/tlbflush.h
+++ b/include/asm-s390/tlbflush.h
@@ -106,9 +106,23 @@ static inline void __tlb_flush_mm_cond(struct mm_struct * mm)
  */
 #define flush_tlb()                             do { } while (0)
 #define flush_tlb_all()                         do { } while (0)
-#define flush_tlb_mm(mm)                        __tlb_flush_mm_cond(mm)
 #define flush_tlb_page(vma, addr)               do { } while (0)
-#define flush_tlb_range(vma, start, end)        __tlb_flush_mm_cond(mm)
-#define flush_tlb_kernel_range(start, end)      __tlb_flush_mm(&init_mm)
+
+static inline void flush_tlb_mm(struct mm_struct *mm)
+{
+        __tlb_flush_mm_cond(mm);
+}
+
+static inline void flush_tlb_range(struct vm_area_struct *vma,
+                                   unsigned long start, unsigned long end)
+{
+        __tlb_flush_mm_cond(vma->vm_mm);
+}
+
+static inline void flush_tlb_kernel_range(unsigned long start,
+                                          unsigned long end)
+{
+        __tlb_flush_mm(&init_mm);
+}
 
 #endif /* _S390_TLBFLUSH_H */