#define EN_FEW_WAIT_VAL                (0x8 << 16)
 #define CLK_DIS_WAIT_VAL       (0x2 << 12)
 
+#define RETAIN_MEM             BIT(14)
+#define RETAIN_PERIPH          BIT(13)
+
 #define TIMEOUT_US             100
 
 #define domain_to_gdsc(domain) container_of(domain, struct gdsc, pd)
        return -ETIMEDOUT;
 }
 
+static inline void gdsc_force_mem_on(struct gdsc *sc)
+{
+       int i;
+       u32 mask = RETAIN_MEM | RETAIN_PERIPH;
+
+       for (i = 0; i < sc->cxc_count; i++)
+               regmap_update_bits(sc->regmap, sc->cxcs[i], mask, mask);
+}
+
+static inline void gdsc_clear_mem_on(struct gdsc *sc)
+{
+       int i;
+       u32 mask = RETAIN_MEM | RETAIN_PERIPH;
+
+       for (i = 0; i < sc->cxc_count; i++)
+               regmap_update_bits(sc->regmap, sc->cxcs[i], mask, 0);
+}
+
 static int gdsc_enable(struct generic_pm_domain *domain)
 {
        struct gdsc *sc = domain_to_gdsc(domain);
        ret = gdsc_toggle_logic(sc, true);
        if (ret)
                return ret;
+
+       if (sc->pwrsts & PWRSTS_OFF)
+               gdsc_force_mem_on(sc);
+
        /*
         * If clocks to this power domain were already on, they will take an
         * additional 4 clock cycles to re-enable after the power domain is
 {
        struct gdsc *sc = domain_to_gdsc(domain);
 
+       if (sc->pwrsts & PWRSTS_OFF)
+               gdsc_clear_mem_on(sc);
+
        return gdsc_toggle_logic(sc, false);
 }
 
        if (on < 0)
                return on;
 
+       if (on || (sc->pwrsts & PWRSTS_RET))
+               gdsc_force_mem_on(sc);
+       else
+               gdsc_clear_mem_on(sc);
+
        sc->pd.power_off = gdsc_disable;
        sc->pd.power_on = gdsc_enable;
        pm_genpd_init(&sc->pd, NULL, !on);
 
 
 struct regmap;
 
+/* Powerdomain allowable state bitfields */
+#define PWRSTS_OFF             BIT(0)
+#define PWRSTS_RET             BIT(1)
+#define PWRSTS_ON              BIT(2)
+#define PWRSTS_OFF_ON          (PWRSTS_OFF | PWRSTS_ON)
+#define PWRSTS_RET_ON          (PWRSTS_RET | PWRSTS_ON)
+
 /**
  * struct gdsc - Globally Distributed Switch Controller
  * @pd: generic power domain
  * @regmap: regmap for MMIO accesses
  * @gdscr: gsdc control register
+ * @cxcs: offsets of branch registers to toggle mem/periph bits in
+ * @cxc_count: number of @cxcs
+ * @pwrsts: Possible powerdomain power states
  */
 struct gdsc {
        struct generic_pm_domain        pd;
        struct regmap                   *regmap;
        unsigned int                    gdscr;
+       unsigned int                    *cxcs;
+       unsigned int                    cxc_count;
+       const u8                        pwrsts;
 };
 
 #ifdef CONFIG_QCOM_GDSC