CM_IDLEST1, CM_IDLEST2, OMAP2430_CM_IDLEST3
 };
 
-
 u32 cm_read_mod_reg(s16 module, u16 idx)
 {
        return __raw_readl(cm_base + module + idx);
        return (i < MAX_MODULE_READY_TIME) ? 0 : -EBUSY;
 }
 
+/*
+ * Context save/restore code - OMAP3 only
+ */
+#ifdef CONFIG_ARCH_OMAP3
+struct omap3_cm_regs {
+       u32 iva2_cm_clksel1;
+       u32 iva2_cm_clksel2;
+       u32 cm_sysconfig;
+       u32 sgx_cm_clksel;
+       u32 dss_cm_clksel;
+       u32 cam_cm_clksel;
+       u32 per_cm_clksel;
+       u32 emu_cm_clksel;
+       u32 emu_cm_clkstctrl;
+       u32 pll_cm_autoidle2;
+       u32 pll_cm_clksel4;
+       u32 pll_cm_clksel5;
+       u32 pll_cm_clken2;
+       u32 cm_polctrl;
+       u32 iva2_cm_fclken;
+       u32 iva2_cm_clken_pll;
+       u32 core_cm_fclken1;
+       u32 core_cm_fclken3;
+       u32 sgx_cm_fclken;
+       u32 wkup_cm_fclken;
+       u32 dss_cm_fclken;
+       u32 cam_cm_fclken;
+       u32 per_cm_fclken;
+       u32 usbhost_cm_fclken;
+       u32 core_cm_iclken1;
+       u32 core_cm_iclken2;
+       u32 core_cm_iclken3;
+       u32 sgx_cm_iclken;
+       u32 wkup_cm_iclken;
+       u32 dss_cm_iclken;
+       u32 cam_cm_iclken;
+       u32 per_cm_iclken;
+       u32 usbhost_cm_iclken;
+       u32 iva2_cm_autoidle2;
+       u32 mpu_cm_autoidle2;
+       u32 iva2_cm_clkstctrl;
+       u32 mpu_cm_clkstctrl;
+       u32 core_cm_clkstctrl;
+       u32 sgx_cm_clkstctrl;
+       u32 dss_cm_clkstctrl;
+       u32 cam_cm_clkstctrl;
+       u32 per_cm_clkstctrl;
+       u32 neon_cm_clkstctrl;
+       u32 usbhost_cm_clkstctrl;
+       u32 core_cm_autoidle1;
+       u32 core_cm_autoidle2;
+       u32 core_cm_autoidle3;
+       u32 wkup_cm_autoidle;
+       u32 dss_cm_autoidle;
+       u32 cam_cm_autoidle;
+       u32 per_cm_autoidle;
+       u32 usbhost_cm_autoidle;
+       u32 sgx_cm_sleepdep;
+       u32 dss_cm_sleepdep;
+       u32 cam_cm_sleepdep;
+       u32 per_cm_sleepdep;
+       u32 usbhost_cm_sleepdep;
+       u32 cm_clkout_ctrl;
+};
+
+static struct omap3_cm_regs cm_context;
+
+void omap3_cm_save_context(void)
+{
+       cm_context.iva2_cm_clksel1 =
+               cm_read_mod_reg(OMAP3430_IVA2_MOD, CM_CLKSEL1);
+       cm_context.iva2_cm_clksel2 =
+               cm_read_mod_reg(OMAP3430_IVA2_MOD, CM_CLKSEL2);
+       cm_context.cm_sysconfig = __raw_readl(OMAP3430_CM_SYSCONFIG);
+       cm_context.sgx_cm_clksel =
+               cm_read_mod_reg(OMAP3430ES2_SGX_MOD, CM_CLKSEL);
+       cm_context.dss_cm_clksel =
+               cm_read_mod_reg(OMAP3430_DSS_MOD, CM_CLKSEL);
+       cm_context.cam_cm_clksel =
+               cm_read_mod_reg(OMAP3430_CAM_MOD, CM_CLKSEL);
+       cm_context.per_cm_clksel =
+               cm_read_mod_reg(OMAP3430_PER_MOD, CM_CLKSEL);
+       cm_context.emu_cm_clksel =
+               cm_read_mod_reg(OMAP3430_EMU_MOD, CM_CLKSEL1);
+       cm_context.emu_cm_clkstctrl =
+               cm_read_mod_reg(OMAP3430_EMU_MOD, OMAP2_CM_CLKSTCTRL);
+       cm_context.pll_cm_autoidle2 =
+               cm_read_mod_reg(PLL_MOD, CM_AUTOIDLE2);
+       cm_context.pll_cm_clksel4 =
+               cm_read_mod_reg(PLL_MOD, OMAP3430ES2_CM_CLKSEL4);
+       cm_context.pll_cm_clksel5 =
+               cm_read_mod_reg(PLL_MOD, OMAP3430ES2_CM_CLKSEL5);
+       cm_context.pll_cm_clken2 =
+               cm_read_mod_reg(PLL_MOD, OMAP3430ES2_CM_CLKEN2);
+       cm_context.cm_polctrl = __raw_readl(OMAP3430_CM_POLCTRL);
+       cm_context.iva2_cm_fclken =
+               cm_read_mod_reg(OMAP3430_IVA2_MOD, CM_FCLKEN);
+       cm_context.iva2_cm_clken_pll = cm_read_mod_reg(OMAP3430_IVA2_MOD,
+                                                      OMAP3430_CM_CLKEN_PLL);
+       cm_context.core_cm_fclken1 =
+               cm_read_mod_reg(CORE_MOD, CM_FCLKEN1);
+       cm_context.core_cm_fclken3 =
+               cm_read_mod_reg(CORE_MOD, OMAP3430ES2_CM_FCLKEN3);
+       cm_context.sgx_cm_fclken =
+               cm_read_mod_reg(OMAP3430ES2_SGX_MOD, CM_FCLKEN);
+       cm_context.wkup_cm_fclken =
+               cm_read_mod_reg(WKUP_MOD, CM_FCLKEN);
+       cm_context.dss_cm_fclken =
+               cm_read_mod_reg(OMAP3430_DSS_MOD, CM_FCLKEN);
+       cm_context.cam_cm_fclken =
+               cm_read_mod_reg(OMAP3430_CAM_MOD, CM_FCLKEN);
+       cm_context.per_cm_fclken =
+               cm_read_mod_reg(OMAP3430_PER_MOD, CM_FCLKEN);
+       cm_context.usbhost_cm_fclken =
+               cm_read_mod_reg(OMAP3430ES2_USBHOST_MOD, CM_FCLKEN);
+       cm_context.core_cm_iclken1 =
+               cm_read_mod_reg(CORE_MOD, CM_ICLKEN1);
+       cm_context.core_cm_iclken2 =
+               cm_read_mod_reg(CORE_MOD, CM_ICLKEN2);
+       cm_context.core_cm_iclken3 =
+               cm_read_mod_reg(CORE_MOD, CM_ICLKEN3);
+       cm_context.sgx_cm_iclken =
+               cm_read_mod_reg(OMAP3430ES2_SGX_MOD, CM_ICLKEN);
+       cm_context.wkup_cm_iclken =
+               cm_read_mod_reg(WKUP_MOD, CM_ICLKEN);
+       cm_context.dss_cm_iclken =
+               cm_read_mod_reg(OMAP3430_DSS_MOD, CM_ICLKEN);
+       cm_context.cam_cm_iclken =
+               cm_read_mod_reg(OMAP3430_CAM_MOD, CM_ICLKEN);
+       cm_context.per_cm_iclken =
+               cm_read_mod_reg(OMAP3430_PER_MOD, CM_ICLKEN);
+       cm_context.usbhost_cm_iclken =
+               cm_read_mod_reg(OMAP3430ES2_USBHOST_MOD, CM_ICLKEN);
+       cm_context.iva2_cm_autoidle2 =
+               cm_read_mod_reg(OMAP3430_IVA2_MOD, CM_AUTOIDLE2);
+       cm_context.mpu_cm_autoidle2 =
+               cm_read_mod_reg(MPU_MOD, CM_AUTOIDLE2);
+       cm_context.iva2_cm_clkstctrl =
+               cm_read_mod_reg(OMAP3430_IVA2_MOD, OMAP2_CM_CLKSTCTRL);
+       cm_context.mpu_cm_clkstctrl =
+               cm_read_mod_reg(MPU_MOD, OMAP2_CM_CLKSTCTRL);
+       cm_context.core_cm_clkstctrl =
+               cm_read_mod_reg(CORE_MOD, OMAP2_CM_CLKSTCTRL);
+       cm_context.sgx_cm_clkstctrl =
+               cm_read_mod_reg(OMAP3430ES2_SGX_MOD, OMAP2_CM_CLKSTCTRL);
+       cm_context.dss_cm_clkstctrl =
+               cm_read_mod_reg(OMAP3430_DSS_MOD, OMAP2_CM_CLKSTCTRL);
+       cm_context.cam_cm_clkstctrl =
+               cm_read_mod_reg(OMAP3430_CAM_MOD, OMAP2_CM_CLKSTCTRL);
+       cm_context.per_cm_clkstctrl =
+               cm_read_mod_reg(OMAP3430_PER_MOD, OMAP2_CM_CLKSTCTRL);
+       cm_context.neon_cm_clkstctrl =
+               cm_read_mod_reg(OMAP3430_NEON_MOD, OMAP2_CM_CLKSTCTRL);
+       cm_context.usbhost_cm_clkstctrl =
+               cm_read_mod_reg(OMAP3430ES2_USBHOST_MOD, OMAP2_CM_CLKSTCTRL);
+       cm_context.core_cm_autoidle1 =
+               cm_read_mod_reg(CORE_MOD, CM_AUTOIDLE1);
+       cm_context.core_cm_autoidle2 =
+               cm_read_mod_reg(CORE_MOD, CM_AUTOIDLE2);
+       cm_context.core_cm_autoidle3 =
+               cm_read_mod_reg(CORE_MOD, CM_AUTOIDLE3);
+       cm_context.wkup_cm_autoidle =
+               cm_read_mod_reg(WKUP_MOD, CM_AUTOIDLE);
+       cm_context.dss_cm_autoidle =
+               cm_read_mod_reg(OMAP3430_DSS_MOD, CM_AUTOIDLE);
+       cm_context.cam_cm_autoidle =
+               cm_read_mod_reg(OMAP3430_CAM_MOD, CM_AUTOIDLE);
+       cm_context.per_cm_autoidle =
+               cm_read_mod_reg(OMAP3430_PER_MOD, CM_AUTOIDLE);
+       cm_context.usbhost_cm_autoidle =
+               cm_read_mod_reg(OMAP3430ES2_USBHOST_MOD, CM_AUTOIDLE);
+       cm_context.sgx_cm_sleepdep =
+               cm_read_mod_reg(OMAP3430ES2_SGX_MOD, OMAP3430_CM_SLEEPDEP);
+       cm_context.dss_cm_sleepdep =
+               cm_read_mod_reg(OMAP3430_DSS_MOD, OMAP3430_CM_SLEEPDEP);
+       cm_context.cam_cm_sleepdep =
+               cm_read_mod_reg(OMAP3430_CAM_MOD, OMAP3430_CM_SLEEPDEP);
+       cm_context.per_cm_sleepdep =
+               cm_read_mod_reg(OMAP3430_PER_MOD, OMAP3430_CM_SLEEPDEP);
+       cm_context.usbhost_cm_sleepdep =
+               cm_read_mod_reg(OMAP3430ES2_USBHOST_MOD, OMAP3430_CM_SLEEPDEP);
+       cm_context.cm_clkout_ctrl =
+               cm_read_mod_reg(OMAP3430_CCR_MOD, OMAP3_CM_CLKOUT_CTRL_OFFSET);
+}
+
+void omap3_cm_restore_context(void)
+{
+       cm_write_mod_reg(cm_context.iva2_cm_clksel1, OMAP3430_IVA2_MOD,
+                        CM_CLKSEL1);
+       cm_write_mod_reg(cm_context.iva2_cm_clksel2, OMAP3430_IVA2_MOD,
+                        CM_CLKSEL2);
+       __raw_writel(cm_context.cm_sysconfig, OMAP3430_CM_SYSCONFIG);
+       cm_write_mod_reg(cm_context.sgx_cm_clksel, OMAP3430ES2_SGX_MOD,
+                        CM_CLKSEL);
+       cm_write_mod_reg(cm_context.dss_cm_clksel, OMAP3430_DSS_MOD,
+                        CM_CLKSEL);
+       cm_write_mod_reg(cm_context.cam_cm_clksel, OMAP3430_CAM_MOD,
+                        CM_CLKSEL);
+       cm_write_mod_reg(cm_context.per_cm_clksel, OMAP3430_PER_MOD,
+                        CM_CLKSEL);
+       cm_write_mod_reg(cm_context.emu_cm_clksel, OMAP3430_EMU_MOD,
+                        CM_CLKSEL1);
+       cm_write_mod_reg(cm_context.emu_cm_clkstctrl, OMAP3430_EMU_MOD,
+                        OMAP2_CM_CLKSTCTRL);
+       cm_write_mod_reg(cm_context.pll_cm_autoidle2, PLL_MOD,
+                        CM_AUTOIDLE2);
+       cm_write_mod_reg(cm_context.pll_cm_clksel4, PLL_MOD,
+                        OMAP3430ES2_CM_CLKSEL4);
+       cm_write_mod_reg(cm_context.pll_cm_clksel5, PLL_MOD,
+                        OMAP3430ES2_CM_CLKSEL5);
+       cm_write_mod_reg(cm_context.pll_cm_clken2, PLL_MOD,
+                        OMAP3430ES2_CM_CLKEN2);
+       __raw_writel(cm_context.cm_polctrl, OMAP3430_CM_POLCTRL);
+       cm_write_mod_reg(cm_context.iva2_cm_fclken, OMAP3430_IVA2_MOD,
+                        CM_FCLKEN);
+       cm_write_mod_reg(cm_context.iva2_cm_clken_pll, OMAP3430_IVA2_MOD,
+                        OMAP3430_CM_CLKEN_PLL);
+       cm_write_mod_reg(cm_context.core_cm_fclken1, CORE_MOD, CM_FCLKEN1);
+       cm_write_mod_reg(cm_context.core_cm_fclken3, CORE_MOD,
+                        OMAP3430ES2_CM_FCLKEN3);
+       cm_write_mod_reg(cm_context.sgx_cm_fclken, OMAP3430ES2_SGX_MOD,
+                        CM_FCLKEN);
+       cm_write_mod_reg(cm_context.wkup_cm_fclken, WKUP_MOD, CM_FCLKEN);
+       cm_write_mod_reg(cm_context.dss_cm_fclken, OMAP3430_DSS_MOD,
+                        CM_FCLKEN);
+       cm_write_mod_reg(cm_context.cam_cm_fclken, OMAP3430_CAM_MOD,
+                        CM_FCLKEN);
+       cm_write_mod_reg(cm_context.per_cm_fclken, OMAP3430_PER_MOD,
+                        CM_FCLKEN);
+       cm_write_mod_reg(cm_context.usbhost_cm_fclken,
+                        OMAP3430ES2_USBHOST_MOD, CM_FCLKEN);
+       cm_write_mod_reg(cm_context.core_cm_iclken1, CORE_MOD, CM_ICLKEN1);
+       cm_write_mod_reg(cm_context.core_cm_iclken2, CORE_MOD, CM_ICLKEN2);
+       cm_write_mod_reg(cm_context.core_cm_iclken3, CORE_MOD, CM_ICLKEN3);
+       cm_write_mod_reg(cm_context.sgx_cm_iclken, OMAP3430ES2_SGX_MOD,
+                        CM_ICLKEN);
+       cm_write_mod_reg(cm_context.wkup_cm_iclken, WKUP_MOD, CM_ICLKEN);
+       cm_write_mod_reg(cm_context.dss_cm_iclken, OMAP3430_DSS_MOD,
+                        CM_ICLKEN);
+       cm_write_mod_reg(cm_context.cam_cm_iclken, OMAP3430_CAM_MOD,
+                        CM_ICLKEN);
+       cm_write_mod_reg(cm_context.per_cm_iclken, OMAP3430_PER_MOD,
+                        CM_ICLKEN);
+       cm_write_mod_reg(cm_context.usbhost_cm_iclken,
+                        OMAP3430ES2_USBHOST_MOD, CM_ICLKEN);
+       cm_write_mod_reg(cm_context.iva2_cm_autoidle2, OMAP3430_IVA2_MOD,
+                        CM_AUTOIDLE2);
+       cm_write_mod_reg(cm_context.mpu_cm_autoidle2, MPU_MOD, CM_AUTOIDLE2);
+       cm_write_mod_reg(cm_context.iva2_cm_clkstctrl, OMAP3430_IVA2_MOD,
+                        OMAP2_CM_CLKSTCTRL);
+       cm_write_mod_reg(cm_context.mpu_cm_clkstctrl, MPU_MOD,
+                        OMAP2_CM_CLKSTCTRL);
+       cm_write_mod_reg(cm_context.core_cm_clkstctrl, CORE_MOD,
+                        OMAP2_CM_CLKSTCTRL);
+       cm_write_mod_reg(cm_context.sgx_cm_clkstctrl, OMAP3430ES2_SGX_MOD,
+                        OMAP2_CM_CLKSTCTRL);
+       cm_write_mod_reg(cm_context.dss_cm_clkstctrl, OMAP3430_DSS_MOD,
+                        OMAP2_CM_CLKSTCTRL);
+       cm_write_mod_reg(cm_context.cam_cm_clkstctrl, OMAP3430_CAM_MOD,
+                        OMAP2_CM_CLKSTCTRL);
+       cm_write_mod_reg(cm_context.per_cm_clkstctrl, OMAP3430_PER_MOD,
+                        OMAP2_CM_CLKSTCTRL);
+       cm_write_mod_reg(cm_context.neon_cm_clkstctrl, OMAP3430_NEON_MOD,
+                        OMAP2_CM_CLKSTCTRL);
+       cm_write_mod_reg(cm_context.usbhost_cm_clkstctrl,
+                        OMAP3430ES2_USBHOST_MOD, OMAP2_CM_CLKSTCTRL);
+       cm_write_mod_reg(cm_context.core_cm_autoidle1, CORE_MOD,
+                        CM_AUTOIDLE1);
+       cm_write_mod_reg(cm_context.core_cm_autoidle2, CORE_MOD,
+                        CM_AUTOIDLE2);
+       cm_write_mod_reg(cm_context.core_cm_autoidle3, CORE_MOD,
+                        CM_AUTOIDLE3);
+       cm_write_mod_reg(cm_context.wkup_cm_autoidle, WKUP_MOD, CM_AUTOIDLE);
+       cm_write_mod_reg(cm_context.dss_cm_autoidle, OMAP3430_DSS_MOD,
+                        CM_AUTOIDLE);
+       cm_write_mod_reg(cm_context.cam_cm_autoidle, OMAP3430_CAM_MOD,
+                        CM_AUTOIDLE);
+       cm_write_mod_reg(cm_context.per_cm_autoidle, OMAP3430_PER_MOD,
+                        CM_AUTOIDLE);
+       cm_write_mod_reg(cm_context.usbhost_cm_autoidle,
+                        OMAP3430ES2_USBHOST_MOD, CM_AUTOIDLE);
+       cm_write_mod_reg(cm_context.sgx_cm_sleepdep, OMAP3430ES2_SGX_MOD,
+                        OMAP3430_CM_SLEEPDEP);
+       cm_write_mod_reg(cm_context.dss_cm_sleepdep, OMAP3430_DSS_MOD,
+                        OMAP3430_CM_SLEEPDEP);
+       cm_write_mod_reg(cm_context.cam_cm_sleepdep, OMAP3430_CAM_MOD,
+                        OMAP3430_CM_SLEEPDEP);
+       cm_write_mod_reg(cm_context.per_cm_sleepdep, OMAP3430_PER_MOD,
+                        OMAP3430_CM_SLEEPDEP);
+       cm_write_mod_reg(cm_context.usbhost_cm_sleepdep,
+                        OMAP3430ES2_USBHOST_MOD, OMAP3430_CM_SLEEPDEP);
+       cm_write_mod_reg(cm_context.cm_clkout_ctrl, OMAP3430_CCR_MOD,
+                        OMAP3_CM_CLKOUT_CTRL_OFFSET);
+}
+#endif
 
 
 #define MAX_MODULE_ENABLE_WAIT         100000
 
-struct omap3_prcm_regs {
-       u32 iva2_cm_clksel1;
-       u32 iva2_cm_clksel2;
-       u32 cm_sysconfig;
-       u32 sgx_cm_clksel;
-       u32 dss_cm_clksel;
-       u32 cam_cm_clksel;
-       u32 per_cm_clksel;
-       u32 emu_cm_clksel;
-       u32 emu_cm_clkstctrl;
-       u32 pll_cm_autoidle2;
-       u32 pll_cm_clksel4;
-       u32 pll_cm_clksel5;
-       u32 pll_cm_clken2;
-       u32 cm_polctrl;
-       u32 iva2_cm_fclken;
-       u32 iva2_cm_clken_pll;
-       u32 core_cm_fclken1;
-       u32 core_cm_fclken3;
-       u32 sgx_cm_fclken;
-       u32 wkup_cm_fclken;
-       u32 dss_cm_fclken;
-       u32 cam_cm_fclken;
-       u32 per_cm_fclken;
-       u32 usbhost_cm_fclken;
-       u32 core_cm_iclken1;
-       u32 core_cm_iclken2;
-       u32 core_cm_iclken3;
-       u32 sgx_cm_iclken;
-       u32 wkup_cm_iclken;
-       u32 dss_cm_iclken;
-       u32 cam_cm_iclken;
-       u32 per_cm_iclken;
-       u32 usbhost_cm_iclken;
-       u32 iva2_cm_autiidle2;
-       u32 mpu_cm_autoidle2;
-       u32 iva2_cm_clkstctrl;
-       u32 mpu_cm_clkstctrl;
-       u32 core_cm_clkstctrl;
-       u32 sgx_cm_clkstctrl;
-       u32 dss_cm_clkstctrl;
-       u32 cam_cm_clkstctrl;
-       u32 per_cm_clkstctrl;
-       u32 neon_cm_clkstctrl;
-       u32 usbhost_cm_clkstctrl;
-       u32 core_cm_autoidle1;
-       u32 core_cm_autoidle2;
-       u32 core_cm_autoidle3;
-       u32 wkup_cm_autoidle;
-       u32 dss_cm_autoidle;
-       u32 cam_cm_autoidle;
-       u32 per_cm_autoidle;
-       u32 usbhost_cm_autoidle;
-       u32 sgx_cm_sleepdep;
-       u32 dss_cm_sleepdep;
-       u32 cam_cm_sleepdep;
-       u32 per_cm_sleepdep;
-       u32 usbhost_cm_sleepdep;
-       u32 cm_clkout_ctrl;
-       u32 prm_clkout_ctrl;
-       u32 sgx_pm_wkdep;
-       u32 dss_pm_wkdep;
-       u32 cam_pm_wkdep;
-       u32 per_pm_wkdep;
-       u32 neon_pm_wkdep;
-       u32 usbhost_pm_wkdep;
-       u32 core_pm_mpugrpsel1;
-       u32 iva2_pm_ivagrpsel1;
-       u32 core_pm_mpugrpsel3;
-       u32 core_pm_ivagrpsel3;
-       u32 wkup_pm_mpugrpsel;
-       u32 wkup_pm_ivagrpsel;
-       u32 per_pm_mpugrpsel;
-       u32 per_pm_ivagrpsel;
-       u32 wkup_pm_wken;
-};
-
-static struct omap3_prcm_regs prcm_context;
-
 u32 omap_prcm_get_reset_sources(void)
 {
        /* XXX This presumably needs modification for 34XX */
                WARN_ON(!cm2_base);
        }
 }
-
-#ifdef CONFIG_ARCH_OMAP3
-void omap3_prcm_save_context(void)
-{
-       prcm_context.iva2_cm_clksel1 =
-                        cm_read_mod_reg(OMAP3430_IVA2_MOD, CM_CLKSEL1);
-       prcm_context.iva2_cm_clksel2 =
-                        cm_read_mod_reg(OMAP3430_IVA2_MOD, CM_CLKSEL2);
-       prcm_context.cm_sysconfig = __raw_readl(OMAP3430_CM_SYSCONFIG);
-       prcm_context.sgx_cm_clksel =
-                        cm_read_mod_reg(OMAP3430ES2_SGX_MOD, CM_CLKSEL);
-       prcm_context.dss_cm_clksel =
-                        cm_read_mod_reg(OMAP3430_DSS_MOD, CM_CLKSEL);
-       prcm_context.cam_cm_clksel =
-                        cm_read_mod_reg(OMAP3430_CAM_MOD, CM_CLKSEL);
-       prcm_context.per_cm_clksel =
-                        cm_read_mod_reg(OMAP3430_PER_MOD, CM_CLKSEL);
-       prcm_context.emu_cm_clksel =
-                        cm_read_mod_reg(OMAP3430_EMU_MOD, CM_CLKSEL1);
-       prcm_context.emu_cm_clkstctrl =
-                        cm_read_mod_reg(OMAP3430_EMU_MOD, OMAP2_CM_CLKSTCTRL);
-       prcm_context.pll_cm_autoidle2 =
-                        cm_read_mod_reg(PLL_MOD, CM_AUTOIDLE2);
-       prcm_context.pll_cm_clksel4 =
-                       cm_read_mod_reg(PLL_MOD, OMAP3430ES2_CM_CLKSEL4);
-       prcm_context.pll_cm_clksel5 =
-                        cm_read_mod_reg(PLL_MOD, OMAP3430ES2_CM_CLKSEL5);
-       prcm_context.pll_cm_clken2 =
-                       cm_read_mod_reg(PLL_MOD, OMAP3430ES2_CM_CLKEN2);
-       prcm_context.cm_polctrl = __raw_readl(OMAP3430_CM_POLCTRL);
-       prcm_context.iva2_cm_fclken =
-                        cm_read_mod_reg(OMAP3430_IVA2_MOD, CM_FCLKEN);
-       prcm_context.iva2_cm_clken_pll = cm_read_mod_reg(OMAP3430_IVA2_MOD,
-                       OMAP3430_CM_CLKEN_PLL);
-       prcm_context.core_cm_fclken1 =
-                        cm_read_mod_reg(CORE_MOD, CM_FCLKEN1);
-       prcm_context.core_cm_fclken3 =
-                        cm_read_mod_reg(CORE_MOD, OMAP3430ES2_CM_FCLKEN3);
-       prcm_context.sgx_cm_fclken =
-                        cm_read_mod_reg(OMAP3430ES2_SGX_MOD, CM_FCLKEN);
-       prcm_context.wkup_cm_fclken =
-                        cm_read_mod_reg(WKUP_MOD, CM_FCLKEN);
-       prcm_context.dss_cm_fclken =
-                        cm_read_mod_reg(OMAP3430_DSS_MOD, CM_FCLKEN);
-       prcm_context.cam_cm_fclken =
-                        cm_read_mod_reg(OMAP3430_CAM_MOD, CM_FCLKEN);
-       prcm_context.per_cm_fclken =
-                        cm_read_mod_reg(OMAP3430_PER_MOD, CM_FCLKEN);
-       prcm_context.usbhost_cm_fclken =
-                        cm_read_mod_reg(OMAP3430ES2_USBHOST_MOD, CM_FCLKEN);
-       prcm_context.core_cm_iclken1 =
-                        cm_read_mod_reg(CORE_MOD, CM_ICLKEN1);
-       prcm_context.core_cm_iclken2 =
-                        cm_read_mod_reg(CORE_MOD, CM_ICLKEN2);
-       prcm_context.core_cm_iclken3 =
-                        cm_read_mod_reg(CORE_MOD, CM_ICLKEN3);
-       prcm_context.sgx_cm_iclken =
-                        cm_read_mod_reg(OMAP3430ES2_SGX_MOD, CM_ICLKEN);
-       prcm_context.wkup_cm_iclken =
-                        cm_read_mod_reg(WKUP_MOD, CM_ICLKEN);
-       prcm_context.dss_cm_iclken =
-                        cm_read_mod_reg(OMAP3430_DSS_MOD, CM_ICLKEN);
-       prcm_context.cam_cm_iclken =
-                        cm_read_mod_reg(OMAP3430_CAM_MOD, CM_ICLKEN);
-       prcm_context.per_cm_iclken =
-                        cm_read_mod_reg(OMAP3430_PER_MOD, CM_ICLKEN);
-       prcm_context.usbhost_cm_iclken =
-                        cm_read_mod_reg(OMAP3430ES2_USBHOST_MOD, CM_ICLKEN);
-       prcm_context.iva2_cm_autiidle2 =
-                        cm_read_mod_reg(OMAP3430_IVA2_MOD, CM_AUTOIDLE2);
-       prcm_context.mpu_cm_autoidle2 =
-                        cm_read_mod_reg(MPU_MOD, CM_AUTOIDLE2);
-       prcm_context.iva2_cm_clkstctrl =
-                        cm_read_mod_reg(OMAP3430_IVA2_MOD, OMAP2_CM_CLKSTCTRL);
-       prcm_context.mpu_cm_clkstctrl =
-                        cm_read_mod_reg(MPU_MOD, OMAP2_CM_CLKSTCTRL);
-       prcm_context.core_cm_clkstctrl =
-                        cm_read_mod_reg(CORE_MOD, OMAP2_CM_CLKSTCTRL);
-       prcm_context.sgx_cm_clkstctrl =
-                        cm_read_mod_reg(OMAP3430ES2_SGX_MOD,
-                                               OMAP2_CM_CLKSTCTRL);
-       prcm_context.dss_cm_clkstctrl =
-                        cm_read_mod_reg(OMAP3430_DSS_MOD, OMAP2_CM_CLKSTCTRL);
-       prcm_context.cam_cm_clkstctrl =
-                        cm_read_mod_reg(OMAP3430_CAM_MOD, OMAP2_CM_CLKSTCTRL);
-       prcm_context.per_cm_clkstctrl =
-                        cm_read_mod_reg(OMAP3430_PER_MOD, OMAP2_CM_CLKSTCTRL);
-       prcm_context.neon_cm_clkstctrl =
-                        cm_read_mod_reg(OMAP3430_NEON_MOD, OMAP2_CM_CLKSTCTRL);
-       prcm_context.usbhost_cm_clkstctrl =
-                        cm_read_mod_reg(OMAP3430ES2_USBHOST_MOD,
-                                               OMAP2_CM_CLKSTCTRL);
-       prcm_context.core_cm_autoidle1 =
-                        cm_read_mod_reg(CORE_MOD, CM_AUTOIDLE1);
-       prcm_context.core_cm_autoidle2 =
-                        cm_read_mod_reg(CORE_MOD, CM_AUTOIDLE2);
-       prcm_context.core_cm_autoidle3 =
-                        cm_read_mod_reg(CORE_MOD, CM_AUTOIDLE3);
-       prcm_context.wkup_cm_autoidle =
-                        cm_read_mod_reg(WKUP_MOD, CM_AUTOIDLE);
-       prcm_context.dss_cm_autoidle =
-                        cm_read_mod_reg(OMAP3430_DSS_MOD, CM_AUTOIDLE);
-       prcm_context.cam_cm_autoidle =
-                        cm_read_mod_reg(OMAP3430_CAM_MOD, CM_AUTOIDLE);
-       prcm_context.per_cm_autoidle =
-                        cm_read_mod_reg(OMAP3430_PER_MOD, CM_AUTOIDLE);
-       prcm_context.usbhost_cm_autoidle =
-                        cm_read_mod_reg(OMAP3430ES2_USBHOST_MOD, CM_AUTOIDLE);
-       prcm_context.sgx_cm_sleepdep =
-                cm_read_mod_reg(OMAP3430ES2_SGX_MOD, OMAP3430_CM_SLEEPDEP);
-       prcm_context.dss_cm_sleepdep =
-                cm_read_mod_reg(OMAP3430_DSS_MOD, OMAP3430_CM_SLEEPDEP);
-       prcm_context.cam_cm_sleepdep =
-                cm_read_mod_reg(OMAP3430_CAM_MOD, OMAP3430_CM_SLEEPDEP);
-       prcm_context.per_cm_sleepdep =
-                cm_read_mod_reg(OMAP3430_PER_MOD, OMAP3430_CM_SLEEPDEP);
-       prcm_context.usbhost_cm_sleepdep =
-                cm_read_mod_reg(OMAP3430ES2_USBHOST_MOD, OMAP3430_CM_SLEEPDEP);
-       prcm_context.cm_clkout_ctrl = cm_read_mod_reg(OMAP3430_CCR_MOD,
-                OMAP3_CM_CLKOUT_CTRL_OFFSET);
-       prcm_context.prm_clkout_ctrl = prm_read_mod_reg(OMAP3430_CCR_MOD,
-               OMAP3_PRM_CLKOUT_CTRL_OFFSET);
-       prcm_context.sgx_pm_wkdep =
-                prm_read_mod_reg(OMAP3430ES2_SGX_MOD, PM_WKDEP);
-       prcm_context.dss_pm_wkdep =
-                prm_read_mod_reg(OMAP3430_DSS_MOD, PM_WKDEP);
-       prcm_context.cam_pm_wkdep =
-                prm_read_mod_reg(OMAP3430_CAM_MOD, PM_WKDEP);
-       prcm_context.per_pm_wkdep =
-                prm_read_mod_reg(OMAP3430_PER_MOD, PM_WKDEP);
-       prcm_context.neon_pm_wkdep =
-                prm_read_mod_reg(OMAP3430_NEON_MOD, PM_WKDEP);
-       prcm_context.usbhost_pm_wkdep =
-                prm_read_mod_reg(OMAP3430ES2_USBHOST_MOD, PM_WKDEP);
-       prcm_context.core_pm_mpugrpsel1 =
-                prm_read_mod_reg(CORE_MOD, OMAP3430_PM_MPUGRPSEL1);
-       prcm_context.iva2_pm_ivagrpsel1 =
-                prm_read_mod_reg(OMAP3430_IVA2_MOD, OMAP3430_PM_IVAGRPSEL1);
-       prcm_context.core_pm_mpugrpsel3 =
-                prm_read_mod_reg(CORE_MOD, OMAP3430ES2_PM_MPUGRPSEL3);
-       prcm_context.core_pm_ivagrpsel3 =
-                prm_read_mod_reg(CORE_MOD, OMAP3430ES2_PM_IVAGRPSEL3);
-       prcm_context.wkup_pm_mpugrpsel =
-                prm_read_mod_reg(WKUP_MOD, OMAP3430_PM_MPUGRPSEL);
-       prcm_context.wkup_pm_ivagrpsel =
-                prm_read_mod_reg(WKUP_MOD, OMAP3430_PM_IVAGRPSEL);
-       prcm_context.per_pm_mpugrpsel =
-                prm_read_mod_reg(OMAP3430_PER_MOD, OMAP3430_PM_MPUGRPSEL);
-       prcm_context.per_pm_ivagrpsel =
-                prm_read_mod_reg(OMAP3430_PER_MOD, OMAP3430_PM_IVAGRPSEL);
-       prcm_context.wkup_pm_wken = prm_read_mod_reg(WKUP_MOD, PM_WKEN);
-       return;
-}
-
-void omap3_prcm_restore_context(void)
-{
-       cm_write_mod_reg(prcm_context.iva2_cm_clksel1, OMAP3430_IVA2_MOD,
-                                        CM_CLKSEL1);
-       cm_write_mod_reg(prcm_context.iva2_cm_clksel2, OMAP3430_IVA2_MOD,
-                                        CM_CLKSEL2);
-       __raw_writel(prcm_context.cm_sysconfig, OMAP3430_CM_SYSCONFIG);
-       cm_write_mod_reg(prcm_context.sgx_cm_clksel, OMAP3430ES2_SGX_MOD,
-                                        CM_CLKSEL);
-       cm_write_mod_reg(prcm_context.dss_cm_clksel, OMAP3430_DSS_MOD,
-                                        CM_CLKSEL);
-       cm_write_mod_reg(prcm_context.cam_cm_clksel, OMAP3430_CAM_MOD,
-                                        CM_CLKSEL);
-       cm_write_mod_reg(prcm_context.per_cm_clksel, OMAP3430_PER_MOD,
-                                        CM_CLKSEL);
-       cm_write_mod_reg(prcm_context.emu_cm_clksel, OMAP3430_EMU_MOD,
-                                        CM_CLKSEL1);
-       cm_write_mod_reg(prcm_context.emu_cm_clkstctrl, OMAP3430_EMU_MOD,
-                                        OMAP2_CM_CLKSTCTRL);
-       cm_write_mod_reg(prcm_context.pll_cm_autoidle2, PLL_MOD,
-                                        CM_AUTOIDLE2);
-       cm_write_mod_reg(prcm_context.pll_cm_clksel4, PLL_MOD,
-                                       OMAP3430ES2_CM_CLKSEL4);
-       cm_write_mod_reg(prcm_context.pll_cm_clksel5, PLL_MOD,
-                                        OMAP3430ES2_CM_CLKSEL5);
-       cm_write_mod_reg(prcm_context.pll_cm_clken2, PLL_MOD,
-                                       OMAP3430ES2_CM_CLKEN2);
-       __raw_writel(prcm_context.cm_polctrl, OMAP3430_CM_POLCTRL);
-       cm_write_mod_reg(prcm_context.iva2_cm_fclken, OMAP3430_IVA2_MOD,
-                                        CM_FCLKEN);
-       cm_write_mod_reg(prcm_context.iva2_cm_clken_pll, OMAP3430_IVA2_MOD,
-                                       OMAP3430_CM_CLKEN_PLL);
-       cm_write_mod_reg(prcm_context.core_cm_fclken1, CORE_MOD, CM_FCLKEN1);
-       cm_write_mod_reg(prcm_context.core_cm_fclken3, CORE_MOD,
-                                        OMAP3430ES2_CM_FCLKEN3);
-       cm_write_mod_reg(prcm_context.sgx_cm_fclken, OMAP3430ES2_SGX_MOD,
-                                        CM_FCLKEN);
-       cm_write_mod_reg(prcm_context.wkup_cm_fclken, WKUP_MOD, CM_FCLKEN);
-       cm_write_mod_reg(prcm_context.dss_cm_fclken, OMAP3430_DSS_MOD,
-                                        CM_FCLKEN);
-       cm_write_mod_reg(prcm_context.cam_cm_fclken, OMAP3430_CAM_MOD,
-                                        CM_FCLKEN);
-       cm_write_mod_reg(prcm_context.per_cm_fclken, OMAP3430_PER_MOD,
-                                        CM_FCLKEN);
-       cm_write_mod_reg(prcm_context.usbhost_cm_fclken,
-                                        OMAP3430ES2_USBHOST_MOD, CM_FCLKEN);
-       cm_write_mod_reg(prcm_context.core_cm_iclken1, CORE_MOD, CM_ICLKEN1);
-       cm_write_mod_reg(prcm_context.core_cm_iclken2, CORE_MOD, CM_ICLKEN2);
-       cm_write_mod_reg(prcm_context.core_cm_iclken3, CORE_MOD, CM_ICLKEN3);
-       cm_write_mod_reg(prcm_context.sgx_cm_iclken, OMAP3430ES2_SGX_MOD,
-                                       CM_ICLKEN);
-       cm_write_mod_reg(prcm_context.wkup_cm_iclken, WKUP_MOD, CM_ICLKEN);
-       cm_write_mod_reg(prcm_context.dss_cm_iclken, OMAP3430_DSS_MOD,
-                                       CM_ICLKEN);
-       cm_write_mod_reg(prcm_context.cam_cm_iclken, OMAP3430_CAM_MOD,
-                                       CM_ICLKEN);
-       cm_write_mod_reg(prcm_context.per_cm_iclken, OMAP3430_PER_MOD,
-                                       CM_ICLKEN);
-       cm_write_mod_reg(prcm_context.usbhost_cm_iclken,
-                                       OMAP3430ES2_USBHOST_MOD, CM_ICLKEN);
-       cm_write_mod_reg(prcm_context.iva2_cm_autiidle2, OMAP3430_IVA2_MOD,
-                                       CM_AUTOIDLE2);
-       cm_write_mod_reg(prcm_context.mpu_cm_autoidle2, MPU_MOD, CM_AUTOIDLE2);
-       cm_write_mod_reg(prcm_context.iva2_cm_clkstctrl, OMAP3430_IVA2_MOD,
-                                       OMAP2_CM_CLKSTCTRL);
-       cm_write_mod_reg(prcm_context.mpu_cm_clkstctrl, MPU_MOD,
-                                       OMAP2_CM_CLKSTCTRL);
-       cm_write_mod_reg(prcm_context.core_cm_clkstctrl, CORE_MOD,
-                                       OMAP2_CM_CLKSTCTRL);
-       cm_write_mod_reg(prcm_context.sgx_cm_clkstctrl, OMAP3430ES2_SGX_MOD,
-                                       OMAP2_CM_CLKSTCTRL);
-       cm_write_mod_reg(prcm_context.dss_cm_clkstctrl, OMAP3430_DSS_MOD,
-                                       OMAP2_CM_CLKSTCTRL);
-       cm_write_mod_reg(prcm_context.cam_cm_clkstctrl, OMAP3430_CAM_MOD,
-                                       OMAP2_CM_CLKSTCTRL);
-       cm_write_mod_reg(prcm_context.per_cm_clkstctrl, OMAP3430_PER_MOD,
-                                       OMAP2_CM_CLKSTCTRL);
-       cm_write_mod_reg(prcm_context.neon_cm_clkstctrl, OMAP3430_NEON_MOD,
-                                       OMAP2_CM_CLKSTCTRL);
-       cm_write_mod_reg(prcm_context.usbhost_cm_clkstctrl,
-                               OMAP3430ES2_USBHOST_MOD, OMAP2_CM_CLKSTCTRL);
-       cm_write_mod_reg(prcm_context.core_cm_autoidle1, CORE_MOD,
-                                       CM_AUTOIDLE1);
-       cm_write_mod_reg(prcm_context.core_cm_autoidle2, CORE_MOD,
-                                       CM_AUTOIDLE2);
-       cm_write_mod_reg(prcm_context.core_cm_autoidle3, CORE_MOD,
-                                       CM_AUTOIDLE3);
-       cm_write_mod_reg(prcm_context.wkup_cm_autoidle, WKUP_MOD, CM_AUTOIDLE);
-       cm_write_mod_reg(prcm_context.dss_cm_autoidle, OMAP3430_DSS_MOD,
-                                       CM_AUTOIDLE);
-       cm_write_mod_reg(prcm_context.cam_cm_autoidle, OMAP3430_CAM_MOD,
-                                       CM_AUTOIDLE);
-       cm_write_mod_reg(prcm_context.per_cm_autoidle, OMAP3430_PER_MOD,
-                                       CM_AUTOIDLE);
-       cm_write_mod_reg(prcm_context.usbhost_cm_autoidle,
-                                       OMAP3430ES2_USBHOST_MOD, CM_AUTOIDLE);
-       cm_write_mod_reg(prcm_context.sgx_cm_sleepdep, OMAP3430ES2_SGX_MOD,
-                                       OMAP3430_CM_SLEEPDEP);
-       cm_write_mod_reg(prcm_context.dss_cm_sleepdep, OMAP3430_DSS_MOD,
-                                       OMAP3430_CM_SLEEPDEP);
-       cm_write_mod_reg(prcm_context.cam_cm_sleepdep, OMAP3430_CAM_MOD,
-                                       OMAP3430_CM_SLEEPDEP);
-       cm_write_mod_reg(prcm_context.per_cm_sleepdep, OMAP3430_PER_MOD,
-                                       OMAP3430_CM_SLEEPDEP);
-       cm_write_mod_reg(prcm_context.usbhost_cm_sleepdep,
-                               OMAP3430ES2_USBHOST_MOD, OMAP3430_CM_SLEEPDEP);
-       cm_write_mod_reg(prcm_context.cm_clkout_ctrl, OMAP3430_CCR_MOD,
-                                       OMAP3_CM_CLKOUT_CTRL_OFFSET);
-       prm_write_mod_reg(prcm_context.prm_clkout_ctrl, OMAP3430_CCR_MOD,
-                                       OMAP3_PRM_CLKOUT_CTRL_OFFSET);
-       prm_write_mod_reg(prcm_context.sgx_pm_wkdep, OMAP3430ES2_SGX_MOD,
-                                       PM_WKDEP);
-       prm_write_mod_reg(prcm_context.dss_pm_wkdep, OMAP3430_DSS_MOD,
-                                       PM_WKDEP);
-       prm_write_mod_reg(prcm_context.cam_pm_wkdep, OMAP3430_CAM_MOD,
-                                       PM_WKDEP);
-       prm_write_mod_reg(prcm_context.per_pm_wkdep, OMAP3430_PER_MOD,
-                                       PM_WKDEP);
-       prm_write_mod_reg(prcm_context.neon_pm_wkdep, OMAP3430_NEON_MOD,
-                                       PM_WKDEP);
-       prm_write_mod_reg(prcm_context.usbhost_pm_wkdep,
-                                       OMAP3430ES2_USBHOST_MOD, PM_WKDEP);
-       prm_write_mod_reg(prcm_context.core_pm_mpugrpsel1, CORE_MOD,
-                                       OMAP3430_PM_MPUGRPSEL1);
-       prm_write_mod_reg(prcm_context.iva2_pm_ivagrpsel1, OMAP3430_IVA2_MOD,
-                                       OMAP3430_PM_IVAGRPSEL1);
-       prm_write_mod_reg(prcm_context.core_pm_mpugrpsel3, CORE_MOD,
-                                       OMAP3430ES2_PM_MPUGRPSEL3);
-       prm_write_mod_reg(prcm_context.core_pm_ivagrpsel3, CORE_MOD,
-                                       OMAP3430ES2_PM_IVAGRPSEL3);
-       prm_write_mod_reg(prcm_context.wkup_pm_mpugrpsel, WKUP_MOD,
-                                       OMAP3430_PM_MPUGRPSEL);
-       prm_write_mod_reg(prcm_context.wkup_pm_ivagrpsel, WKUP_MOD,
-                                       OMAP3430_PM_IVAGRPSEL);
-       prm_write_mod_reg(prcm_context.per_pm_mpugrpsel, OMAP3430_PER_MOD,
-                                       OMAP3430_PM_MPUGRPSEL);
-       prm_write_mod_reg(prcm_context.per_pm_ivagrpsel, OMAP3430_PER_MOD,
-                                        OMAP3430_PM_IVAGRPSEL);
-       prm_write_mod_reg(prcm_context.wkup_pm_wken, WKUP_MOD, PM_WKEN);
-       return;
-}
-#endif