u64                     sctlr;
 };
 
-static void __tlb_switch_to_guest(struct kvm_s2_mmu *mmu,
-                                 struct tlb_inv_context *cxt)
+static void enter_vmid_context(struct kvm_s2_mmu *mmu,
+                              struct tlb_inv_context *cxt)
 {
        struct kvm_vcpu *vcpu = kvm_get_running_vcpu();
        u64 val;
        isb();
 }
 
-static void __tlb_switch_to_host(struct tlb_inv_context *cxt)
+static void exit_vmid_context(struct tlb_inv_context *cxt)
 {
        /*
         * We're done with the TLB operation, let's restore the host's
        dsb(ishst);
 
        /* Switch to requested VMID */
-       __tlb_switch_to_guest(mmu, &cxt);
+       enter_vmid_context(mmu, &cxt);
 
        /*
         * We could do so much better if we had the VA as well.
        dsb(ish);
        isb();
 
-       __tlb_switch_to_host(&cxt);
+       exit_vmid_context(&cxt);
 }
 
 void __kvm_tlb_flush_vmid_ipa_nsh(struct kvm_s2_mmu *mmu,
        dsb(nshst);
 
        /* Switch to requested VMID */
-       __tlb_switch_to_guest(mmu, &cxt);
+       enter_vmid_context(mmu, &cxt);
 
        /*
         * We could do so much better if we had the VA as well.
        dsb(nsh);
        isb();
 
-       __tlb_switch_to_host(&cxt);
+       exit_vmid_context(&cxt);
 }
 
 void __kvm_tlb_flush_vmid_range(struct kvm_s2_mmu *mmu,
        dsb(ishst);
 
        /* Switch to requested VMID */
-       __tlb_switch_to_guest(mmu, &cxt);
+       enter_vmid_context(mmu, &cxt);
 
        __flush_s2_tlb_range_op(ipas2e1is, start, pages, stride, 0);
 
        dsb(ish);
        isb();
 
-       __tlb_switch_to_host(&cxt);
+       exit_vmid_context(&cxt);
 }
 
 void __kvm_tlb_flush_vmid(struct kvm_s2_mmu *mmu)
        dsb(ishst);
 
        /* Switch to requested VMID */
-       __tlb_switch_to_guest(mmu, &cxt);
+       enter_vmid_context(mmu, &cxt);
 
        __tlbi(vmalls12e1is);
        dsb(ish);
        isb();
 
-       __tlb_switch_to_host(&cxt);
+       exit_vmid_context(&cxt);
 }
 
 void __kvm_flush_cpu_context(struct kvm_s2_mmu *mmu)
        struct tlb_inv_context cxt;
 
        /* Switch to requested VMID */
-       __tlb_switch_to_guest(mmu, &cxt);
+       enter_vmid_context(mmu, &cxt);
 
        __tlbi(vmalle1);
        asm volatile("ic iallu");
        dsb(nsh);
        isb();
 
-       __tlb_switch_to_host(&cxt);
+       exit_vmid_context(&cxt);
 }
 
 void __kvm_flush_vm_context(void)