return sof_ipc3_ctx_ipc(sdev, SOF_IPC_PM_CTX_RESTORE);
 }
 
+static int sof_ipc3_set_pm_gate(struct snd_sof_dev *sdev, u32 flags)
+{
+       struct sof_ipc_pm_gate pm_gate;
+       struct sof_ipc_reply reply;
+
+       memset(&pm_gate, 0, sizeof(pm_gate));
+
+       /* configure pm_gate ipc message */
+       pm_gate.hdr.size = sizeof(pm_gate);
+       pm_gate.hdr.cmd = SOF_IPC_GLB_PM_MSG | SOF_IPC_PM_GATE;
+       pm_gate.flags = flags;
+
+       /* send pm_gate ipc to dsp */
+       return sof_ipc_tx_message_no_pm(sdev->ipc, &pm_gate, sizeof(pm_gate),
+                                       &reply, sizeof(reply));
+}
+
 static const struct sof_ipc_pm_ops ipc3_pm_ops = {
        .ctx_save = sof_ipc3_ctx_save,
        .ctx_restore = sof_ipc3_ctx_restore,
        .set_core_state = sof_ipc3_set_core_state,
+       .set_pm_gate = sof_ipc3_set_pm_gate,
 };
 
 const struct sof_ipc_ops ipc3_ops = {
 
        return sof_ipc4_set_core_state(sdev, SOF_DSP_PRIMARY_CORE, false);
 }
 
+static int sof_ipc4_set_pm_gate(struct snd_sof_dev *sdev, u32 flags)
+{
+       struct sof_ipc4_msg msg = {{0}};
+
+       msg.primary = SOF_IPC4_MSG_TYPE_SET(SOF_IPC4_MOD_SET_D0IX);
+       msg.primary |= SOF_IPC4_MSG_DIR(SOF_IPC4_MSG_REQUEST);
+       msg.primary |= SOF_IPC4_MSG_TARGET(SOF_IPC4_MODULE_MSG);
+       msg.extension = flags;
+
+       return sof_ipc4_tx_msg(sdev, &msg, 0, NULL, 0, true);
+}
+
 static const struct sof_ipc_pm_ops ipc4_pm_ops = {
        .ctx_save = sof_ipc4_ctx_save,
        .set_core_state = sof_ipc4_set_core_state,
+       .set_pm_gate = sof_ipc4_set_pm_gate,
 };
 
 static int sof_ipc4_init(struct snd_sof_dev *sdev)
 
  * @ctx_save:          Optional function pointer for context save
  * @ctx_restore:       Optional function pointer for context restore
  * @set_core_state:    Optional function pointer for turning on/off a DSP core
+ * @set_pm_gate:       Optional function pointer for pm gate settings
  */
 struct sof_ipc_pm_ops {
        int (*ctx_save)(struct snd_sof_dev *sdev);
        int (*ctx_restore)(struct snd_sof_dev *sdev);
        int (*set_core_state)(struct snd_sof_dev *sdev, int core_idx, bool on);
+       int (*set_pm_gate)(struct snd_sof_dev *sdev, u32 flags);
 };
 
 /**