#define HCLGEVF_NAME   "hclgevf"
 
+static int hclgevf_init_hdev(struct hclgevf_dev *hdev);
+static void hclgevf_uninit_hdev(struct hclgevf_dev *hdev);
 static struct hnae3_ae_algo ae_algovf;
 
 static const struct pci_device_id ae_algovf_pci_tbl[] = {
        struct hclgevf_tqp *tqp;
        int i;
 
+       /* if this is on going reset then we need to re-allocate the TPQs
+        * since we cannot assume we would get same number of TPQs back from PF
+        */
+       if (hclgevf_dev_ongoing_reset(hdev))
+               devm_kfree(&hdev->pdev->dev, hdev->htqp);
+
        hdev->htqp = devm_kcalloc(&hdev->pdev->dev, hdev->num_tqps,
                                  sizeof(struct hclgevf_tqp), GFP_KERNEL);
        if (!hdev->htqp)
        new_tqps = kinfo->rss_size * kinfo->num_tc;
        kinfo->num_tqps = min(new_tqps, hdev->num_tqps);
 
+       /* if this is on going reset then we need to re-allocate the hnae queues
+        * as well since number of TPQs from PF might have changed.
+        */
+       if (hclgevf_dev_ongoing_reset(hdev))
+               devm_kfree(&hdev->pdev->dev, kinfo->tqp);
+
        kinfo->tqp = devm_kcalloc(&hdev->pdev->dev, kinfo->num_tqps,
                                  sizeof(struct hnae3_queue *), GFP_KERNEL);
        if (!kinfo->tqp)
 
 static int hclgevf_reset_stack(struct hclgevf_dev *hdev)
 {
+       int ret;
+
        /* uninitialize the nic client */
        hclgevf_notify_client(hdev, HNAE3_UNINIT_CLIENT);
 
-       /* re-initialize the hclge device - add code here */
+       /* re-initialize the hclge device */
+       ret = hclgevf_init_hdev(hdev);
+       if (ret) {
+               dev_err(&hdev->pdev->dev,
+                       "hclge device re-init failed, VF is disabled!\n");
+               return ret;
+       }
 
        /* bring up the nic client again */
        hclgevf_notify_client(hdev, HNAE3_INIT_CLIENT);
        return hclgevf_get_tc_info(hdev);
 }
 
+static int hclgevf_alloc_hdev(struct hnae3_ae_dev *ae_dev)
+{
+       struct pci_dev *pdev = ae_dev->pdev;
+       struct hclgevf_dev *hdev = ae_dev->priv;
+
+       hdev = devm_kzalloc(&pdev->dev, sizeof(*hdev), GFP_KERNEL);
+       if (!hdev)
+               return -ENOMEM;
+
+       hdev->pdev = pdev;
+       hdev->ae_dev = ae_dev;
+       ae_dev->priv = hdev;
+
+       return 0;
+}
+
 static int hclgevf_init_roce_base_info(struct hclgevf_dev *hdev)
 {
        struct hnae3_handle *roce = &hdev->roce;
 
 static void hclgevf_state_init(struct hclgevf_dev *hdev)
 {
+       /* if this is on going reset then skip this initialization */
+       if (hclgevf_dev_ongoing_reset(hdev))
+               return;
+
        /* setup tasks for the MBX */
        INIT_WORK(&hdev->mbx_service_task, hclgevf_mailbox_service_task);
        clear_bit(HCLGEVF_STATE_MBX_SERVICE_SCHED, &hdev->state);
        int vectors;
        int i;
 
+       /* if this is on going reset then skip this initialization */
+       if (hclgevf_dev_ongoing_reset(hdev))
+               return 0;
+
        hdev->num_msi = HCLGEVF_MAX_VF_VECTOR_NUM;
 
        vectors = pci_alloc_irq_vectors(pdev, 1, hdev->num_msi,
 {
        int ret = 0;
 
+       /* if this is on going reset then skip this initialization */
+       if (hclgevf_dev_ongoing_reset(hdev))
+               return 0;
+
        hclgevf_get_misc_vector(hdev);
 
        ret = request_irq(hdev->misc_vector.vector_irq, hclgevf_misc_irq_handle,
        struct hclgevf_hw *hw;
        int ret;
 
+       /* check if we need to skip initialization of pci. This will happen if
+        * device is undergoing VF reset. Otherwise, we would need to
+        * re-initialize pci interface again i.e. when device is not going
+        * through *any* reset or actually undergoing full reset.
+        */
+       if (hclgevf_dev_ongoing_reset(hdev))
+               return 0;
+
        ret = pci_enable_device(pdev);
        if (ret) {
                dev_err(&pdev->dev, "failed to enable PCI device\n");
        pci_set_drvdata(pdev, NULL);
 }
 
-static int hclgevf_init_ae_dev(struct hnae3_ae_dev *ae_dev)
+static int hclgevf_init_hdev(struct hclgevf_dev *hdev)
 {
-       struct pci_dev *pdev = ae_dev->pdev;
-       struct hclgevf_dev *hdev;
+       struct pci_dev *pdev = hdev->pdev;
        int ret;
 
-       hdev = devm_kzalloc(&pdev->dev, sizeof(*hdev), GFP_KERNEL);
-       if (!hdev)
-               return -ENOMEM;
-
-       hdev->pdev = pdev;
-       hdev->ae_dev = ae_dev;
-       ae_dev->priv = hdev;
+       /* check if device is on-going full reset(i.e. pcie as well) */
+       if (hclgevf_dev_ongoing_full_reset(hdev)) {
+               dev_warn(&pdev->dev, "device is going full reset\n");
+               hclgevf_uninit_hdev(hdev);
+       }
 
        ret = hclgevf_pci_init(hdev);
        if (ret) {
        return ret;
 }
 
-static void hclgevf_uninit_ae_dev(struct hnae3_ae_dev *ae_dev)
+static void hclgevf_uninit_hdev(struct hclgevf_dev *hdev)
 {
-       struct hclgevf_dev *hdev = ae_dev->priv;
-
        hclgevf_cmd_uninit(hdev);
        hclgevf_misc_irq_uninit(hdev);
        hclgevf_state_uninit(hdev);
        hclgevf_uninit_msi(hdev);
        hclgevf_pci_uninit(hdev);
+}
+
+static int hclgevf_init_ae_dev(struct hnae3_ae_dev *ae_dev)
+{
+       struct pci_dev *pdev = ae_dev->pdev;
+       int ret;
+
+       ret = hclgevf_alloc_hdev(ae_dev);
+       if (ret) {
+               dev_err(&pdev->dev, "hclge device allocation failed\n");
+               return ret;
+       }
+
+       ret = hclgevf_init_hdev(ae_dev->priv);
+       if (ret)
+               dev_err(&pdev->dev, "hclge device initialization failed\n");
+
+       return ret;
+}
+
+static void hclgevf_uninit_ae_dev(struct hnae3_ae_dev *ae_dev)
+{
+       struct hclgevf_dev *hdev = ae_dev->priv;
+
+       hclgevf_uninit_hdev(hdev);
        ae_dev->priv = NULL;
 }