mpi3mr_cleanup_resources(mrioc);
 }
 
-#ifdef CONFIG_PM
 /**
  * mpi3mr_suspend - PCI power management suspend callback
- * @pdev: PCI device instance
- * @state: New power state
+ * @dev: Device struct
  *
  * Change the power state to the given value and cleanup the IOC
  * by issuing MUR and shutdown notification
  *
  * Return: 0 always.
  */
-static int mpi3mr_suspend(struct pci_dev *pdev, pm_message_t state)
+static int __maybe_unused
+mpi3mr_suspend(struct device *dev)
 {
+       struct pci_dev *pdev = to_pci_dev(dev);
        struct Scsi_Host *shost = pci_get_drvdata(pdev);
        struct mpi3mr_ioc *mrioc;
-       pci_power_t device_state;
 
        if (!shost)
                return 0;
        mpi3mr_stop_watchdog(mrioc);
        mpi3mr_cleanup_ioc(mrioc);
 
-       device_state = pci_choose_state(pdev, state);
-       ioc_info(mrioc, "pdev=0x%p, slot=%s, entering operating state [D%d]\n",
-           pdev, pci_name(pdev), device_state);
-       pci_save_state(pdev);
+       ioc_info(mrioc, "pdev=0x%p, slot=%s, entering operating state\n",
+           pdev, pci_name(pdev));
        mpi3mr_cleanup_resources(mrioc);
-       pci_set_power_state(pdev, device_state);
 
        return 0;
 }
 
 /**
  * mpi3mr_resume - PCI power management resume callback
- * @pdev: PCI device instance
+ * @dev: Device struct
  *
  * Restore the power state to D0 and reinitialize the controller
  * and resume I/O operations to the target devices
  *
  * Return: 0 on success, non-zero on failure
  */
-static int mpi3mr_resume(struct pci_dev *pdev)
+static int __maybe_unused
+mpi3mr_resume(struct device *dev)
 {
+       struct pci_dev *pdev = to_pci_dev(dev);
        struct Scsi_Host *shost = pci_get_drvdata(pdev);
        struct mpi3mr_ioc *mrioc;
        pci_power_t device_state = pdev->current_state;
 
        ioc_info(mrioc, "pdev=0x%p, slot=%s, previous operating state [D%d]\n",
            pdev, pci_name(pdev), device_state);
-       pci_set_power_state(pdev, PCI_D0);
-       pci_enable_wake(pdev, PCI_D0, 0);
-       pci_restore_state(pdev);
        mrioc->pdev = pdev;
        mrioc->cpu_count = num_online_cpus();
        r = mpi3mr_setup_resources(mrioc);
 
        return 0;
 }
-#endif
 
 static const struct pci_device_id mpi3mr_pci_id_table[] = {
        {
 };
 MODULE_DEVICE_TABLE(pci, mpi3mr_pci_id_table);
 
+static SIMPLE_DEV_PM_OPS(mpi3mr_pm_ops, mpi3mr_suspend, mpi3mr_resume);
+
 static struct pci_driver mpi3mr_pci_driver = {
        .name = MPI3MR_DRIVER_NAME,
        .id_table = mpi3mr_pci_id_table,
        .probe = mpi3mr_probe,
        .remove = mpi3mr_remove,
        .shutdown = mpi3mr_shutdown,
-#ifdef CONFIG_PM
-       .suspend = mpi3mr_suspend,
-       .resume = mpi3mr_resume,
-#endif
+       .driver.pm = &mpi3mr_pm_ops,
 };
 
 static ssize_t event_counter_show(struct device_driver *dd, char *buf)