x86/mm: Only use tlb_remove_table() for paravirt

[linux.git] / arch / x86 / hyperv / mmu.c

diff --git a/arch/x86/hyperv/mmu.c b/arch/x86/hyperv/mmu.c
index 1147e1fed7fff45e28e9c18e59e81542d1647d7f..ef5f29f913d7b064f1a086ac0674bebb4c8a845a 100644 (file)
--- a/arch/x86/hyperv/mmu.c
+++ b/arch/x86/hyperv/mmu.c
@@ -9,6 +9,7 @@
 #include <asm/mshyperv.h>
 #include <asm/msr.h>
 #include <asm/tlbflush.h>
+#include <asm/tlb.h>
 
 #define CREATE_TRACE_POINTS
 #include <asm/trace/hyperv.h>
@@ -231,4 +232,5 @@ void hyperv_setup_mmu_ops(void)
 
        pr_info("Using hypercall for remote TLB flush\n");
        pv_mmu_ops.flush_tlb_others = hyperv_flush_tlb_others;
+       pv_mmu_ops.tlb_remove_table = tlb_remove_table;
 }