ANDROID: KVM: arm64: Fix merge memslot-ops into pkvm-allocator
diff --git a/arch/arm64/kvm/hyp/nvhe/pkvm.c b/arch/arm64/kvm/hyp/nvhe/pkvm.c
index 6443cfa..424f4b3 100644
--- a/arch/arm64/kvm/hyp/nvhe/pkvm.c
+++ b/arch/arm64/kvm/hyp/nvhe/pkvm.c
@@ -377,7 +377,7 @@ struct pkvm_hyp_vm *get_pkvm_hyp_vm(pkvm_handle_t handle)
if (hyp_vm->is_dying)
hyp_vm = NULL;
else
- hyp_page_ref_inc(hyp_virt_to_page(hyp_vm));
+ hyp_refcount_inc(hyp_vm->refcount);
unlock:
hyp_read_unlock(&vm_table_lock);
@@ -387,7 +387,7 @@ struct pkvm_hyp_vm *get_pkvm_hyp_vm(pkvm_handle_t handle)
void put_pkvm_hyp_vm(struct pkvm_hyp_vm *hyp_vm)
{
- hyp_page_ref_dec(hyp_virt_to_page(hyp_vm));
+ hyp_refcount_dec(hyp_vm->refcount);
}
int __pkvm_reclaim_dying_guest_page(pkvm_handle_t handle, u64 pfn, u64 ipa)
