Add runtime suspend/resume for the AIE4 classic device by using its
system suspend/resume handlers.

Runtime suspend returns -EBUSY for a device without runtime_suspend,
so the PM core keeps it active instead of entering an error state.

Wrap .driver.pm with pm_ptr() to allow the dev_pm_ops structure and
its callbacks to be discarded as dead code when CONFIG_PM is disabled.

Signed-off-by: David Zhang <[email protected]>
---
 drivers/accel/amdxdna/aie2_pci.c        |  2 ++
 drivers/accel/amdxdna/aie4_pci.c        |  4 +++
 drivers/accel/amdxdna/amdxdna_pci_drv.c |  4 +--
 drivers/accel/amdxdna/amdxdna_pci_drv.h |  2 ++
 drivers/accel/amdxdna/amdxdna_pm.c      | 34 +++++++++++++++++++++++++
 drivers/accel/amdxdna/amdxdna_pm.h      |  2 ++
 6 files changed, 46 insertions(+), 2 deletions(-)

diff --git a/drivers/accel/amdxdna/aie2_pci.c b/drivers/accel/amdxdna/aie2_pci.c
index 3d9ddd40908d..7f56514c5e26 100644
--- a/drivers/accel/amdxdna/aie2_pci.c
+++ b/drivers/accel/amdxdna/aie2_pci.c
@@ -1220,6 +1220,8 @@ const struct amdxdna_dev_ops aie2_ops = {
        .fini = aie2_fini,
        .resume = aie2_hw_resume,
        .suspend = aie2_hw_suspend,
+       .runtime_resume = aie2_hw_resume,
+       .runtime_suspend = aie2_hw_suspend,
        .get_aie_info = aie2_get_info,
        .set_aie_state = aie2_set_state,
        .hwctx_init = aie2_hwctx_init,
diff --git a/drivers/accel/amdxdna/aie4_pci.c b/drivers/accel/amdxdna/aie4_pci.c
index a368bd3c257f..ee64a5f3405d 100644
--- a/drivers/accel/amdxdna/aie4_pci.c
+++ b/drivers/accel/amdxdna/aie4_pci.c
@@ -1040,6 +1040,7 @@ static int aie4_classic_init(struct amdxdna_dev *xdna)
        if (ret)
                goto free_work_buf;
 
+       amdxdna_pm_init(xdna);
        return 0;
 
 free_work_buf:
@@ -1063,6 +1064,7 @@ static void aie4_vf_fini(struct amdxdna_dev *xdna)
 
 static void aie4_classic_fini(struct amdxdna_dev *xdna)
 {
+       amdxdna_pm_fini(xdna);
        aie4_classic_hw_stop(xdna->dev_handle);
        aie4_free_fw_log_buffer(xdna->dev_handle);
        aie4_free_work_buffer(xdna->dev_handle);
@@ -1403,4 +1405,6 @@ const struct amdxdna_dev_ops aie4_classic_ops = {
        .set_aie_state          = aie4_set_state,
        .resume                 = aie4_classic_resume,
        .suspend                = aie4_classic_suspend,
+       .runtime_resume         = aie4_classic_resume,
+       .runtime_suspend        = aie4_classic_suspend,
 };
diff --git a/drivers/accel/amdxdna/amdxdna_pci_drv.c 
b/drivers/accel/amdxdna/amdxdna_pci_drv.c
index f5f7831c4e80..cb78f145ffa1 100644
--- a/drivers/accel/amdxdna/amdxdna_pci_drv.c
+++ b/drivers/accel/amdxdna/amdxdna_pci_drv.c
@@ -465,7 +465,7 @@ static void amdxdna_remove(struct pci_dev *pdev)
 
 static const struct dev_pm_ops amdxdna_pm_ops = {
        SYSTEM_SLEEP_PM_OPS(amdxdna_pm_suspend, amdxdna_pm_resume)
-       RUNTIME_PM_OPS(amdxdna_pm_suspend, amdxdna_pm_resume, NULL)
+       RUNTIME_PM_OPS(amdxdna_pm_runtime_suspend, amdxdna_pm_runtime_resume, 
NULL)
 };
 
 static int amdxdna_sriov_configure(struct pci_dev *pdev, int num_vfs)
@@ -484,7 +484,7 @@ static struct pci_driver amdxdna_pci_driver = {
        .id_table = pci_ids,
        .probe = amdxdna_probe,
        .remove = amdxdna_remove,
-       .driver.pm = &amdxdna_pm_ops,
+       .driver.pm = pm_ptr(&amdxdna_pm_ops),
        .sriov_configure = amdxdna_sriov_configure,
 };
 
diff --git a/drivers/accel/amdxdna/amdxdna_pci_drv.h 
b/drivers/accel/amdxdna/amdxdna_pci_drv.h
index 505477b4a292..4d047cf3d981 100644
--- a/drivers/accel/amdxdna/amdxdna_pci_drv.h
+++ b/drivers/accel/amdxdna/amdxdna_pci_drv.h
@@ -57,6 +57,8 @@ struct amdxdna_dev_ops {
        void (*debugfs_init)(struct amdxdna_dev *xdna);
        int (*resume)(struct amdxdna_dev *xdna);
        int (*suspend)(struct amdxdna_dev *xdna);
+       int (*runtime_resume)(struct amdxdna_dev *xdna);
+       int (*runtime_suspend)(struct amdxdna_dev *xdna);
        int (*sriov_configure)(struct amdxdna_dev *xdna, int num_vfs);
        int (*hwctx_init)(struct amdxdna_hwctx *hwctx);
        void (*hwctx_stop)(struct amdxdna_hwctx *hwctx);
diff --git a/drivers/accel/amdxdna/amdxdna_pm.c 
b/drivers/accel/amdxdna/amdxdna_pm.c
index b1fafddd7ad5..7d7bc965342d 100644
--- a/drivers/accel/amdxdna/amdxdna_pm.c
+++ b/drivers/accel/amdxdna/amdxdna_pm.c
@@ -37,11 +37,40 @@ int amdxdna_pm_resume(struct device *dev)
        return ret;
 }
 
+int amdxdna_pm_runtime_suspend(struct device *dev)
+{
+       struct amdxdna_dev *xdna = to_xdna_dev(dev_get_drvdata(dev));
+       int ret = -EBUSY;
+
+       guard(mutex)(&xdna->dev_lock);
+       if (xdna->dev_info->ops->runtime_suspend)
+               ret = xdna->dev_info->ops->runtime_suspend(xdna);
+
+       XDNA_DBG(xdna, "Runtime suspend done ret %d", ret);
+       return ret;
+}
+
+int amdxdna_pm_runtime_resume(struct device *dev)
+{
+       struct amdxdna_dev *xdna = to_xdna_dev(dev_get_drvdata(dev));
+       int ret = -EOPNOTSUPP;
+
+       guard(mutex)(&xdna->dev_lock);
+       if (xdna->dev_info->ops->runtime_resume)
+               ret = xdna->dev_info->ops->runtime_resume(xdna);
+
+       XDNA_DBG(xdna, "Runtime resume done ret %d", ret);
+       return ret;
+}
+
 int amdxdna_pm_resume_get(struct amdxdna_dev *xdna)
 {
        struct device *dev = xdna->ddev.dev;
        int ret;
 
+       if (!pm_runtime_enabled(dev))
+               return 0;
+
        ret = pm_runtime_resume_and_get(dev);
        if (ret) {
                XDNA_ERR(xdna, "Resume failed: %d", ret);
@@ -55,6 +84,10 @@ void amdxdna_pm_suspend_put(struct amdxdna_dev *xdna)
 {
        struct device *dev = xdna->ddev.dev;
 
+       if (!pm_runtime_enabled(dev))
+               return;
+
+       pm_runtime_mark_last_busy(dev);
        pm_runtime_put_autosuspend(dev);
 }
 
@@ -66,6 +99,7 @@ void amdxdna_pm_init(struct amdxdna_dev *xdna)
        pm_runtime_set_autosuspend_delay(dev, AMDXDNA_AUTOSUSPEND_DELAY);
        pm_runtime_use_autosuspend(dev);
        pm_runtime_allow(dev);
+       pm_runtime_mark_last_busy(dev);
        pm_runtime_put_autosuspend(dev);
 }
 
diff --git a/drivers/accel/amdxdna/amdxdna_pm.h 
b/drivers/accel/amdxdna/amdxdna_pm.h
index 3d26b973e0e3..02beef6448f1 100644
--- a/drivers/accel/amdxdna/amdxdna_pm.h
+++ b/drivers/accel/amdxdna/amdxdna_pm.h
@@ -10,6 +10,8 @@
 
 int amdxdna_pm_suspend(struct device *dev);
 int amdxdna_pm_resume(struct device  *dev);
+int amdxdna_pm_runtime_suspend(struct device *dev);
+int amdxdna_pm_runtime_resume(struct device *dev);
 int amdxdna_pm_resume_get(struct amdxdna_dev *xdna);
 void amdxdna_pm_suspend_put(struct amdxdna_dev *xdna);
 void amdxdna_pm_init(struct amdxdna_dev *xdna);
-- 
2.34.1

Reply via email to