if (hdmi->i2c)
                dw_hdmi_i2c_init(hdmi);
 
-       platform_set_drvdata(pdev, hdmi);
-
        return hdmi;
 
 err_iahb:
 /* -----------------------------------------------------------------------------
  * Probe/remove API, used from platforms based on the DRM bridge API.
  */
-int dw_hdmi_probe(struct platform_device *pdev,
-                 const struct dw_hdmi_plat_data *plat_data)
+struct dw_hdmi *dw_hdmi_probe(struct platform_device *pdev,
+                             const struct dw_hdmi_plat_data *plat_data)
 {
        struct dw_hdmi *hdmi;
 
        hdmi = __dw_hdmi_probe(pdev, plat_data);
        if (IS_ERR(hdmi))
-               return PTR_ERR(hdmi);
+               return hdmi;
 
        drm_bridge_add(&hdmi->bridge);
 
-       return 0;
+       return hdmi;
 }
 EXPORT_SYMBOL_GPL(dw_hdmi_probe);
 
-void dw_hdmi_remove(struct platform_device *pdev)
+void dw_hdmi_remove(struct dw_hdmi *hdmi)
 {
-       struct dw_hdmi *hdmi = platform_get_drvdata(pdev);
-
        drm_bridge_remove(&hdmi->bridge);
 
        __dw_hdmi_remove(hdmi);
 /* -----------------------------------------------------------------------------
  * Bind/unbind API, used from platforms based on the component framework.
  */
-int dw_hdmi_bind(struct platform_device *pdev, struct drm_encoder *encoder,
-                const struct dw_hdmi_plat_data *plat_data)
+struct dw_hdmi *dw_hdmi_bind(struct platform_device *pdev,
+                            struct drm_encoder *encoder,
+                            const struct dw_hdmi_plat_data *plat_data)
 {
        struct dw_hdmi *hdmi;
        int ret;
 
        hdmi = __dw_hdmi_probe(pdev, plat_data);
        if (IS_ERR(hdmi))
-               return PTR_ERR(hdmi);
+               return hdmi;
 
        ret = drm_bridge_attach(encoder, &hdmi->bridge, NULL);
        if (ret) {
-               dw_hdmi_remove(pdev);
+               dw_hdmi_remove(hdmi);
                DRM_ERROR("Failed to initialize bridge with drm\n");
-               return ret;
+               return ERR_PTR(ret);
        }
 
-       return 0;
+       return hdmi;
 }
 EXPORT_SYMBOL_GPL(dw_hdmi_bind);
 
-void dw_hdmi_unbind(struct device *dev)
+void dw_hdmi_unbind(struct dw_hdmi *hdmi)
 {
-       struct dw_hdmi *hdmi = dev_get_drvdata(dev);
-
        __dw_hdmi_remove(hdmi);
 }
 EXPORT_SYMBOL_GPL(dw_hdmi_unbind);
 
 struct imx_hdmi {
        struct device *dev;
        struct drm_encoder encoder;
+       struct dw_hdmi *hdmi;
        struct regmap *regmap;
 };
 
        drm_encoder_init(drm, encoder, &dw_hdmi_imx_encoder_funcs,
                         DRM_MODE_ENCODER_TMDS, NULL);
 
-       ret = dw_hdmi_bind(pdev, encoder, plat_data);
+       platform_set_drvdata(pdev, hdmi);
+
+       hdmi->hdmi = dw_hdmi_bind(pdev, encoder, plat_data);
 
        /*
         * If dw_hdmi_bind() fails we'll never call dw_hdmi_unbind(),
         * which would have called the encoder cleanup.  Do it manually.
         */
-       if (ret)
+       if (IS_ERR(hdmi->hdmi)) {
+               ret = PTR_ERR(hdmi->hdmi);
                drm_encoder_cleanup(encoder);
+       }
 
        return ret;
 }
 static void dw_hdmi_imx_unbind(struct device *dev, struct device *master,
                               void *data)
 {
-       return dw_hdmi_unbind(dev);
+       struct imx_hdmi *hdmi = dev_get_drvdata(dev);
+
+       dw_hdmi_unbind(hdmi->hdmi);
 }
 
 static const struct component_ops dw_hdmi_imx_ops = {
 
        struct clk *venci_clk;
        struct regulator *hdmi_supply;
        u32 irq_stat;
+       struct dw_hdmi *hdmi;
 };
 #define encoder_to_meson_dw_hdmi(x) \
        container_of(x, struct meson_dw_hdmi, encoder)
        dw_plat_data->input_bus_format = MEDIA_BUS_FMT_YUV8_1X24;
        dw_plat_data->input_bus_encoding = V4L2_YCBCR_ENC_709;
 
-       ret = dw_hdmi_bind(pdev, encoder, &meson_dw_hdmi->dw_plat_data);
-       if (ret)
-               return ret;
+       platform_set_drvdata(pdev, meson_dw_hdmi);
+
+       meson_dw_hdmi->hdmi = dw_hdmi_bind(pdev, encoder,
+                                          &meson_dw_hdmi->dw_plat_data);
+       if (IS_ERR(meson_dw_hdmi->hdmi))
+               return PTR_ERR(meson_dw_hdmi->hdmi);
 
        DRM_DEBUG_DRIVER("HDMI controller initialized\n");
 
 static void meson_dw_hdmi_unbind(struct device *dev, struct device *master,
                                   void *data)
 {
-       dw_hdmi_unbind(dev);
+       struct meson_dw_hdmi *meson_dw_hdmi = dev_get_drvdata(dev);
+
+       dw_hdmi_unbind(meson_dw_hdmi->hdmi);
 }
 
 static const struct component_ops meson_dw_hdmi_ops = {
 
 
 static int rcar_dw_hdmi_probe(struct platform_device *pdev)
 {
-       return dw_hdmi_probe(pdev, &rcar_dw_hdmi_plat_data);
+       struct dw_hdmi *hdmi;
+
+       hdmi = dw_hdmi_probe(pdev, &rcar_dw_hdmi_plat_data);
+       if (IS_ERR(hdmi))
+               return PTR_ERR(hdmi);
+
+       platform_set_drvdata(pdev, hdmi);
 }
 
 static int rcar_dw_hdmi_remove(struct platform_device *pdev)
 {
-       dw_hdmi_remove(pdev);
+       struct dw_hdmi *hdmi = platform_get_drvdata(dev);
+
+       dw_hdmi_remove(hdmi);
 
        return 0;
 }
 
        const struct rockchip_hdmi_chip_data *chip_data;
        struct clk *vpll_clk;
        struct clk *grf_clk;
+       struct dw_hdmi *hdmi;
 };
 
 #define to_rockchip_hdmi(x)    container_of(x, struct rockchip_hdmi, x)
        drm_encoder_init(drm, encoder, &dw_hdmi_rockchip_encoder_funcs,
                         DRM_MODE_ENCODER_TMDS, NULL);
 
-       ret = dw_hdmi_bind(pdev, encoder, plat_data);
+       platform_set_drvdata(pdev, hdmi);
+
+       hdmi->hdmi = dw_hdmi_bind(pdev, encoder, plat_data);
 
        /*
         * If dw_hdmi_bind() fails we'll never call dw_hdmi_unbind(),
         * which would have called the encoder cleanup.  Do it manually.
         */
-       if (ret)
+       if (IS_ERR(hdmi->hdmi)) {
+               ret = PTR_ERR(hdmi->hdmi);
                drm_encoder_cleanup(encoder);
+       }
 
        return ret;
 }
 static void dw_hdmi_rockchip_unbind(struct device *dev, struct device *master,
                                    void *data)
 {
-       return dw_hdmi_unbind(dev);
+       struct rockchip_hdmi *hdmi = dev_get_drvdata(dev);
+
+       dw_hdmi_unbind(hdmi->hdmi);
 }
 
 static const struct component_ops dw_hdmi_rockchip_ops = {
 
                             unsigned long mpixelclock);
 };
 
-int dw_hdmi_probe(struct platform_device *pdev,
-                 const struct dw_hdmi_plat_data *plat_data);
-void dw_hdmi_remove(struct platform_device *pdev);
-void dw_hdmi_unbind(struct device *dev);
-int dw_hdmi_bind(struct platform_device *pdev, struct drm_encoder *encoder,
-                const struct dw_hdmi_plat_data *plat_data);
+struct dw_hdmi *dw_hdmi_probe(struct platform_device *pdev,
+                             const struct dw_hdmi_plat_data *plat_data);
+void dw_hdmi_remove(struct dw_hdmi *hdmi);
+void dw_hdmi_unbind(struct dw_hdmi *hdmi);
+struct dw_hdmi *dw_hdmi_bind(struct platform_device *pdev,
+                            struct drm_encoder *encoder,
+                            const struct dw_hdmi_plat_data *plat_data);
 
 void dw_hdmi_setup_rx_sense(struct device *dev, bool hpd, bool rx_sense);