.max_mlo_peer = 256,
 
                .otp_board_id_register = QCN9274_QFPROM_RAW_RFA_PDET_ROW13_LSB,
+
+               .supports_sta_ps = false,
        },
        {
                .name = "wcn7850 hw2.0",
                .max_mlo_peer = 32,
 
                .otp_board_id_register = 0,
+
+               .supports_sta_ps = true,
        },
        {
                .name = "qcn9274 hw2.0",
                .max_mlo_peer = 256,
 
                .otp_board_id_register = QCN9274_QFPROM_RAW_RFA_PDET_ROW13_LSB,
+
+               .supports_sta_ps = false,
        },
 };
 
 
        return ret;
 }
 
+static void ath12k_mac_vif_setup_ps(struct ath12k_vif *arvif)
+{
+       struct ath12k *ar = arvif->ar;
+       struct ieee80211_vif *vif = arvif->vif;
+       struct ieee80211_conf *conf = &ath12k_ar_to_hw(ar)->conf;
+       enum wmi_sta_powersave_param param;
+       enum wmi_sta_ps_mode psmode;
+       int ret;
+       int timeout;
+       bool enable_ps;
+
+       lockdep_assert_held(&ar->conf_mutex);
+
+       if (vif->type != NL80211_IFTYPE_STATION)
+               return;
+
+       enable_ps = arvif->ps;
+       if (enable_ps) {
+               psmode = WMI_STA_PS_MODE_ENABLED;
+               param = WMI_STA_PS_PARAM_INACTIVITY_TIME;
+
+               timeout = conf->dynamic_ps_timeout;
+               if (timeout == 0) {
+                       /* firmware doesn't like 0 */
+                       timeout = ieee80211_tu_to_usec(vif->bss_conf.beacon_int) / 1000;
+               }
+
+               ret = ath12k_wmi_set_sta_ps_param(ar, arvif->vdev_id, param,
+                                                 timeout);
+               if (ret) {
+                       ath12k_warn(ar->ab, "failed to set inactivity time for vdev %d: %i\n",
+                                   arvif->vdev_id, ret);
+                       return;
+               }
+       } else {
+               psmode = WMI_STA_PS_MODE_DISABLED;
+       }
+
+       ath12k_dbg(ar->ab, ATH12K_DBG_MAC, "mac vdev %d psmode %s\n",
+                  arvif->vdev_id, psmode ? "enable" : "disable");
+
+       ret = ath12k_wmi_pdev_set_ps_mode(ar, arvif->vdev_id, psmode);
+       if (ret)
+               ath12k_warn(ar->ab, "failed to set sta power save mode %d for vdev %d: %d\n",
+                           psmode, arvif->vdev_id, ret);
+}
+
 static void ath12k_mac_bss_info_changed(struct ath12k *ar,
                                        struct ath12k_vif *arvif,
                                        struct ieee80211_bss_conf *info,
                                        u64 changed)
 {
        struct ieee80211_vif *vif = arvif->vif;
+       struct ieee80211_vif_cfg *vif_cfg = &vif->cfg;
        struct cfg80211_chan_def def;
        u32 param_id, param_value;
        enum nl80211_band band;
 
        if (changed & BSS_CHANGED_EHT_PUNCTURING)
                arvif->punct_bitmap = info->eht_puncturing;
+
+       if (changed & BSS_CHANGED_PS &&
+           ar->ab->hw_params->supports_sta_ps) {
+               arvif->ps = vif_cfg->ps;
+               ath12k_mac_vif_setup_ps(arvif);
+       }
 }
 
 static void ath12k_mac_op_bss_info_changed(struct ieee80211_hw *hw,