#include "dp_rx.h"
 #include "debug.h"
 #include "hif.h"
+#include "wow.h"
 
 unsigned int ath11k_debug_mask;
 EXPORT_SYMBOL(ath11k_debug_mask);
                .supports_shadow_regs = false,
                .idle_ps = false,
                .cold_boot_calib = true,
+               .supports_suspend = false,
        },
        {
                .hw_rev = ATH11K_HW_IPQ6018_HW10,
                .supports_shadow_regs = false,
                .idle_ps = false,
                .cold_boot_calib = true,
+               .supports_suspend = false,
        },
        {
                .name = "qca6390 hw2.0",
                .supports_shadow_regs = true,
                .idle_ps = true,
                .cold_boot_calib = false,
+               .supports_suspend = true,
        },
 };
 
+int ath11k_core_suspend(struct ath11k_base *ab)
+{
+       int ret;
+
+       if (!ab->hw_params.supports_suspend)
+               return -EOPNOTSUPP;
+
+       /* TODO: there can frames in queues so for now add delay as a hack.
+        * Need to implement to handle and remove this delay.
+        */
+       msleep(500);
+
+       ret = ath11k_dp_rx_pktlog_stop(ab, true);
+       if (ret) {
+               ath11k_warn(ab, "failed to stop dp rx (and timer) pktlog during suspend: %d\n",
+                           ret);
+               return ret;
+       }
+
+       ret = ath11k_wow_enable(ab);
+       if (ret) {
+               ath11k_warn(ab, "failed to enable wow during suspend: %d\n", ret);
+               return ret;
+       }
+
+       ret = ath11k_dp_rx_pktlog_stop(ab, false);
+       if (ret) {
+               ath11k_warn(ab, "failed to stop dp rx pktlog during suspend: %d\n",
+                           ret);
+               return ret;
+       }
+
+       ath11k_ce_stop_shadow_timers(ab);
+       ath11k_dp_stop_shadow_timers(ab);
+
+       ath11k_hif_irq_disable(ab);
+       ath11k_hif_ce_irq_disable(ab);
+
+       ret = ath11k_hif_suspend(ab);
+       if (!ret) {
+               ath11k_warn(ab, "failed to suspend hif: %d\n", ret);
+               return ret;
+       }
+
+       return 0;
+}
+EXPORT_SYMBOL(ath11k_core_suspend);
+
+int ath11k_core_resume(struct ath11k_base *ab)
+{
+       int ret;
+
+       if (!ab->hw_params.supports_suspend)
+               return -EOPNOTSUPP;
+
+       ret = ath11k_hif_resume(ab);
+       if (ret) {
+               ath11k_warn(ab, "failed to resume hif during resume: %d\n", ret);
+               return ret;
+       }
+
+       ath11k_hif_ce_irq_enable(ab);
+       ath11k_hif_irq_enable(ab);
+
+       ret = ath11k_dp_rx_pktlog_start(ab);
+       if (ret) {
+               ath11k_warn(ab, "failed to start rx pktlog during resume: %d\n",
+                           ret);
+               return ret;
+       }
+
+       ret = ath11k_wow_wakeup(ab);
+       if (ret) {
+               ath11k_warn(ab, "failed to wakeup wow during resume: %d\n", ret);
+               return ret;
+       }
+
+       return 0;
+}
+EXPORT_SYMBOL(ath11k_core_resume);
+
 int ath11k_core_check_dt(struct ath11k_base *ab)
 {
        size_t max_len = sizeof(ab->qmi.target.bdf_ext);
 
        ath11k_pci_power_down(ab);
 }
 
+static __maybe_unused int ath11k_pci_pm_suspend(struct device *dev)
+{
+       struct ath11k_base *ab = dev_get_drvdata(dev);
+       int ret;
+
+       ret = ath11k_core_suspend(ab);
+       if (ret)
+               ath11k_warn(ab, "failed to suspend core: %d\n", ret);
+
+       return ret;
+}
+
+static __maybe_unused int ath11k_pci_pm_resume(struct device *dev)
+{
+       struct ath11k_base *ab = dev_get_drvdata(dev);
+       int ret;
+
+       ret = ath11k_core_resume(ab);
+       if (ret)
+               ath11k_warn(ab, "failed to resume core: %d\n", ret);
+
+       return ret;
+}
+
+static SIMPLE_DEV_PM_OPS(ath11k_pci_pm_ops,
+                        ath11k_pci_pm_suspend,
+                        ath11k_pci_pm_resume);
+
 static struct pci_driver ath11k_pci_driver = {
        .name = "ath11k_pci",
        .id_table = ath11k_pci_id_table,
        .probe = ath11k_pci_probe,
        .remove = ath11k_pci_remove,
        .shutdown = ath11k_pci_shutdown,
+#ifdef CONFIG_PM
+       .driver.pm = &ath11k_pci_pm_ops,
+#endif
 };
 
 static int ath11k_pci_init(void)