diff --git a/drivers/infiniband/hw/hns/hns_roce_hw_v2.c b/drivers/infiniband/hw/hns/hns_roce_hw_v2.c
index e905d26bb3e5190fbd91aa0ced5aed2f5b258a29..63715443bd7dc67a33196a8e233de2477ae4955f 100644
--- a/drivers/infiniband/hw/hns/hns_roce_hw_v2.c
+++ b/drivers/infiniband/hw/hns/hns_roce_hw_v2.c
@@ -2740,6 +2740,8 @@ static void free_dip_list(struct hns_roce_dev *hr_dev)
 
 static int hns_roce_v2_get_reset_page(struct hns_roce_dev *hr_dev)
 {
+	struct hns_roce_v2_reset_state *state;
+
 	hr_dev->reset_page = alloc_page(GFP_KERNEL | __GFP_ZERO);
 	if (!hr_dev->reset_page)
 		return -ENOMEM;
@@ -2748,6 +2750,9 @@ static int hns_roce_v2_get_reset_page(struct hns_roce_dev *hr_dev)
 	if (!hr_dev->reset_kaddr)
 		goto err_with_vmap;
 
+	state = hr_dev->reset_kaddr;
+	state->hw_ready = ~state->hw_ready;
+
 	return 0;
 
 err_with_vmap:
@@ -7425,6 +7430,7 @@ static void hns_roce_v2_reset_notify_user(struct hns_roce_dev *hr_dev)
 	state = (struct hns_roce_v2_reset_state *)hr_dev->reset_kaddr;
 
 	state->reset_state = HNS_ROCE_IS_RESETTING;
+	state->hw_ready = 0;
 	/* Ensure reset state was flushed in memory */
 	wmb();
 }
diff --git a/drivers/infiniband/hw/hns/hns_roce_hw_v2.h b/drivers/infiniband/hw/hns/hns_roce_hw_v2.h
index 228aeb2a80eb30f29a7fc1df4663a3e6c0ee345b..f4cb1ed55b3f7dc7dc14871eaea8825fa0977ee8 100644
--- a/drivers/infiniband/hw/hns/hns_roce_hw_v2.h
+++ b/drivers/infiniband/hw/hns/hns_roce_hw_v2.h
@@ -1314,6 +1314,7 @@ struct hns_roce_link_table {
 
 struct hns_roce_v2_reset_state {
 	u32 reset_state; /* stored to use in user space */
+	u32 hw_ready;
 };
 
 struct hns_roce_v2_free_mr {
diff --git a/drivers/infiniband/hw/hns/hns_roce_main.c b/drivers/infiniband/hw/hns/hns_roce_main.c
index 4a16200ab950336b1cf7501db60b7dc55af1dc5e..2f0a5b2bbc503ddc3f3b1cfe173ca4de7fe6f417 100644
--- a/drivers/infiniband/hw/hns/hns_roce_main.c
+++ b/drivers/infiniband/hw/hns/hns_roce_main.c
@@ -675,10 +675,9 @@ static int hns_roce_mmap(struct ib_ucontext *uctx, struct vm_area_struct *vma)
 			goto out;
 		}
 
-		ret = remap_pfn_range(vma, vma->vm_start,
-				      page_to_pfn(hr_dev->reset_page),
-				      PAGE_SIZE, vma->vm_page_prot);
-		goto out;
+		prot = vma->vm_page_prot;
+		pfn = page_to_pfn(hr_dev->reset_page);
+		break;
 	default:
 		ret = -EINVAL;
 		goto out;