int id;
 
        void __iomem *ctrl_base;
+       phys_addr_t ctrl_size;
        struct regulator_bulk_data supplies[DSI_DEV_REGULATOR_MAX];
 
        struct clk *bus_clks[DSI_BUS_CLK_MAX];
                goto fail;
        }
 
-       msm_host->ctrl_base = msm_ioremap(pdev, "dsi_ctrl", "DSI CTRL");
+       msm_host->ctrl_base = msm_ioremap_size(pdev, "dsi_ctrl", "DSI CTRL", &msm_host->ctrl_size);
        if (IS_ERR(msm_host->ctrl_base)) {
                pr_err("%s: unable to map Dsi ctrl base\n", __func__);
                ret = PTR_ERR(msm_host->ctrl_base);
 
        pm_runtime_get_sync(&msm_host->pdev->dev);
 
-       msm_disp_snapshot_add_block(disp_state, msm_iomap_size(msm_host->pdev, "dsi_ctrl"),
+       msm_disp_snapshot_add_block(disp_state, msm_host->ctrl_size,
                        msm_host->ctrl_base, "dsi%d_ctrl", msm_host->id);
 
        pm_runtime_put_sync(&msm_host->pdev->dev);
 
 }
 
 static void __iomem *_msm_ioremap(struct platform_device *pdev, const char *name,
-                                 const char *dbgname, bool quiet)
+                                 const char *dbgname, bool quiet, phys_addr_t *psize)
 {
        struct resource *res;
        unsigned long size;
        if (reglog)
                printk(KERN_DEBUG "IO:region %s %p %08lx\n", dbgname, ptr, size);
 
+       if (psize)
+               *psize = size;
+
        return ptr;
 }
 
 void __iomem *msm_ioremap(struct platform_device *pdev, const char *name,
                          const char *dbgname)
 {
-       return _msm_ioremap(pdev, name, dbgname, false);
+       return _msm_ioremap(pdev, name, dbgname, false, NULL);
 }
 
 void __iomem *msm_ioremap_quiet(struct platform_device *pdev, const char *name,
                                const char *dbgname)
 {
-       return _msm_ioremap(pdev, name, dbgname, true);
+       return _msm_ioremap(pdev, name, dbgname, true, NULL);
 }
 
-unsigned long msm_iomap_size(struct platform_device *pdev, const char *name)
+void __iomem *msm_ioremap_size(struct platform_device *pdev, const char *name,
+                         const char *dbgname, phys_addr_t *psize)
 {
-       struct resource *res;
-
-       if (name)
-               res = platform_get_resource_byname(pdev, IORESOURCE_MEM, name);
-       else
-               res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
-
-       if (!res) {
-               dev_dbg(&pdev->dev, "failed to get memory resource: %s\n",
-                               name);
-               return 0;
-       }
-
-       return resource_size(res);
+       return _msm_ioremap(pdev, name, dbgname, false, psize);
 }
 
 void msm_writel(u32 data, void __iomem *addr)
 
        const char *name);
 void __iomem *msm_ioremap(struct platform_device *pdev, const char *name,
                const char *dbgname);
+void __iomem *msm_ioremap_size(struct platform_device *pdev, const char *name,
+               const char *dbgname, phys_addr_t *size);
 void __iomem *msm_ioremap_quiet(struct platform_device *pdev, const char *name,
                const char *dbgname);
-unsigned long msm_iomap_size(struct platform_device *pdev, const char *name);
 void msm_writel(u32 data, void __iomem *addr);
 u32 msm_readl(const void __iomem *addr);
 void msm_rmw(void __iomem *addr, u32 mask, u32 or);