#include <linux/rwsem.h>
 #include <linux/suspend.h>
 
+#include <soc/rockchip/pm_domains.h>
 #include <soc/rockchip/rk3399_grf.h>
 #include <soc/rockchip/rockchip_sip.h>
 
 
        mutex_lock(&dmcfreq->lock);
 
+       /*
+        * Ensure power-domain transitions don't interfere with ARM Trusted
+        * Firmware power-domain idling.
+        */
+       err = rockchip_pmu_block();
+       if (err) {
+               dev_err(dev, "Failed to block PMU: %d\n", err);
+               goto out_unlock;
+       }
+
        /*
         * Some idle parameters may be based on the DDR controller clock, which
         * is half of the DDR frequency.
        dmcfreq->volt = target_volt;
 
 out:
+       rockchip_pmu_unblock();
+out_unlock:
        mutex_unlock(&dmcfreq->lock);
        return err;
 }