*/
 
 #include <asm/kvm_hyp.h>
+#include <asm/tlbflush.h>
 
 void __hyp_text __kvm_tlb_flush_vmid_ipa(struct kvm *kvm, phys_addr_t ipa)
 {
         * whole of Stage-1. Weep...
         */
        ipa >>= 12;
-       asm volatile("tlbi ipas2e1is, %0" : : "r" (ipa));
+       __tlbi(ipas2e1is, ipa);
 
        /*
         * We have to ensure completion of the invalidation at Stage-2,
         * the Stage-1 invalidation happened first.
         */
        dsb(ish);
-       asm volatile("tlbi vmalle1is" : : );
+       __tlbi(vmalle1is);
        dsb(ish);
        isb();
 
        write_sysreg(kvm->arch.vttbr, vttbr_el2);
        isb();
 
-       asm volatile("tlbi vmalls12e1is" : : );
+       __tlbi(vmalls12e1is);
        dsb(ish);
        isb();
 
        write_sysreg(kvm->arch.vttbr, vttbr_el2);
        isb();
 
-       asm volatile("tlbi vmalle1" : : );
+       __tlbi(vmalle1);
        dsb(nsh);
        isb();
 
 void __hyp_text __kvm_flush_vm_context(void)
 {
        dsb(ishst);
-       asm volatile("tlbi alle1is      \n"
-                    "ic ialluis          ": : );
+       __tlbi(alle1is);
+       asm volatile("ic ialluis" : : );
        dsb(ish);
 }