1 files changed, 8 insertions, 7 deletions

diff --git a/drivers/nvme/host/pci.c b/drivers/nvme/host/pci.c
index fed803232edc..9a3d69b8df98 100644
--- a/drivers/nvme/host/pci.c
+++ b/drivers/nvme/host/pci.c
@@ -2145,14 +2145,14 @@ static int nvme_probe(struct pci_dev *pdev, const struct pci_device_id *id)
         return result;
 }
 
-static void nvme_reset_notify(struct pci_dev *pdev, bool prepare)
+static void nvme_reset_prepare(struct pci_dev *pdev)
 {
-        struct nvme_dev *dev = pci_get_drvdata(pdev);
+        nvme_dev_disable(pci_get_drvdata(pdev), false);
+}
 
-        if (prepare)
-                nvme_dev_disable(dev, false);
-        else
-                nvme_reset(dev);
+static void nvme_reset_done(struct pci_dev *pdev)
+{
+        nvme_reset(pci_get_drvdata(pdev));
 }
 
 static void nvme_shutdown(struct pci_dev *pdev)
@@ -2275,7 +2275,8 @@ static const struct pci_error_handlers nvme_err_handler = {
         .error_detected = nvme_error_detected,
         .slot_reset     = nvme_slot_reset,
         .resume         = nvme_error_resume,
-        .reset_notify   = nvme_reset_notify,
+        .reset_prepare  = nvme_reset_prepare,
+        .reset_done     = nvme_reset_done,
 };
 
 static const struct pci_device_id nvme_id_table[] = {