struct icc_path *path0;
        struct icc_path *path1;
        struct drm_device *dev = dpu_kms->dev;
+       struct device *dpu_dev = dev->dev;
+       struct device *mdss_dev = dpu_dev->parent;
 
-       path0 = of_icc_get(dev->dev, "mdp0-mem");
-       path1 = of_icc_get(dev->dev, "mdp1-mem");
+       /* Interconnects are a part of MDSS device tree binding, not the
+        * MDP/DPU device. */
+       path0 = of_icc_get(mdss_dev, "mdp0-mem");
+       path1 = of_icc_get(mdss_dev, "mdp1-mem");
 
        if (IS_ERR_OR_NULL(path0))
                return PTR_ERR_OR_ZERO(path0);
        _dpu_kms_hw_destroy(dpu_kms);
 
        msm_kms_destroy(&dpu_kms->base);
+
+       if (dpu_kms->rpm_enabled)
+               pm_runtime_disable(&dpu_kms->pdev->dev);
 }
 
 static irqreturn_t dpu_irq(struct msm_kms *kms)
        struct iommu_domain *domain;
        struct msm_gem_address_space *aspace;
        struct msm_mmu *mmu;
+       struct device *dpu_dev = dpu_kms->dev->dev;
+       struct device *mdss_dev = dpu_dev->parent;
 
        domain = iommu_domain_alloc(&platform_bus_type);
        if (!domain)
                return 0;
 
-       mmu = msm_iommu_new(dpu_kms->dev->dev, domain);
+       /* IOMMUs are a part of MDSS device tree binding, not the
+        * MDP/DPU device. */
+       mmu = msm_iommu_new(mdss_dev, domain);
        if (IS_ERR(mmu)) {
                iommu_domain_free(domain);
                return PTR_ERR(mmu);
        return rc;
 }
 
-static int dpu_kms_init(struct drm_device *dev)
-{
-       struct msm_drm_private *priv;
-       struct dpu_kms *dpu_kms;
-       int irq;
-
-       if (!dev) {
-               DPU_ERROR("drm device node invalid\n");
-               return -EINVAL;
-       }
-
-       priv = dev->dev_private;
-       dpu_kms = to_dpu_kms(priv->kms);
-
-       irq = irq_of_parse_and_map(dpu_kms->pdev->dev.of_node, 0);
-       if (irq < 0) {
-               DPU_ERROR("failed to get irq: %d\n", irq);
-               return irq;
-       }
-       dpu_kms->base.irq = irq;
-
-       return 0;
-}
-
-static int dpu_bind(struct device *dev, struct device *master, void *data)
+static int dpu_kms_init(struct drm_device *ddev)
 {
-       struct msm_drm_private *priv = dev_get_drvdata(master);
+       struct msm_drm_private *priv = ddev->dev_private;
+       struct device *dev = ddev->dev;
        struct platform_device *pdev = to_platform_device(dev);
-       struct drm_device *ddev = priv->dev;
        struct dpu_kms *dpu_kms;
+       int irq;
        int ret = 0;
 
-       priv->kms_init = dpu_kms_init;
-
        dpu_kms = devm_kzalloc(&pdev->dev, sizeof(*dpu_kms), GFP_KERNEL);
        if (!dpu_kms)
                return -ENOMEM;
        }
        dpu_kms->num_clocks = ret;
 
-       platform_set_drvdata(pdev, dpu_kms);
-
        ret = msm_kms_init(&dpu_kms->base, &kms_funcs);
        if (ret) {
                DPU_ERROR("failed to init kms, ret=%d\n", ret);
 
        priv->kms = &dpu_kms->base;
 
-       return ret;
-}
-
-static void dpu_unbind(struct device *dev, struct device *master, void *data)
-{
-       struct platform_device *pdev = to_platform_device(dev);
-       struct dpu_kms *dpu_kms = platform_get_drvdata(pdev);
+       irq = irq_of_parse_and_map(dpu_kms->pdev->dev.of_node, 0);
+       if (irq < 0) {
+               DPU_ERROR("failed to get irq: %d\n", irq);
+               return irq;
+       }
+       dpu_kms->base.irq = irq;
 
-       if (dpu_kms->rpm_enabled)
-               pm_runtime_disable(&pdev->dev);
+       return 0;
 }
 
-static const struct component_ops dpu_ops = {
-       .bind   = dpu_bind,
-       .unbind = dpu_unbind,
-};
-
 static int dpu_dev_probe(struct platform_device *pdev)
 {
-       return component_add(&pdev->dev, &dpu_ops);
+       return msm_drv_probe(&pdev->dev, dpu_kms_init);
 }
 
 static int dpu_dev_remove(struct platform_device *pdev)
 {
-       component_del(&pdev->dev, &dpu_ops);
+       component_master_del(&pdev->dev, &msm_drm_ops);
+
        return 0;
 }
 
 {
        int i;
        struct platform_device *pdev = to_platform_device(dev);
-       struct dpu_kms *dpu_kms = platform_get_drvdata(pdev);
+       struct msm_drm_private *priv = platform_get_drvdata(pdev);
+       struct dpu_kms *dpu_kms = to_dpu_kms(priv->kms);
 
        /* Drop the performance state vote */
        dev_pm_opp_set_rate(dev, 0);
 {
        int rc = -1;
        struct platform_device *pdev = to_platform_device(dev);
-       struct dpu_kms *dpu_kms = platform_get_drvdata(pdev);
+       struct msm_drm_private *priv = platform_get_drvdata(pdev);
+       struct dpu_kms *dpu_kms = to_dpu_kms(priv->kms);
        struct drm_encoder *encoder;
        struct drm_device *ddev;
        int i;
        SET_RUNTIME_PM_OPS(dpu_runtime_suspend, dpu_runtime_resume, NULL)
        SET_SYSTEM_SLEEP_PM_OPS(pm_runtime_force_suspend,
                                pm_runtime_force_resume)
+       .prepare = msm_pm_prepare,
+       .complete = msm_pm_complete,
 };
 
-const struct of_device_id dpu_dt_match[] = {
+static const struct of_device_id dpu_dt_match[] = {
        { .compatible = "qcom,msm8998-dpu", },
        { .compatible = "qcom,qcm2290-dpu", },
        { .compatible = "qcom,sdm845-dpu", },
 
 
 static int mdp4_probe(struct platform_device *pdev)
 {
-       struct msm_drm_private *priv;
-
-       priv = devm_kzalloc(&pdev->dev, sizeof(*priv), GFP_KERNEL);
-       if (!priv)
-               return -ENOMEM;
-
-       priv->kms_init = mdp4_kms_init;
-
-       platform_set_drvdata(pdev, priv);
-
-       /*
-        * on MDP4 based platforms, the MDP platform device is the component
-        * that adds other display interface components to itself.
-        */
-       return msm_drv_probe(&pdev->dev, &pdev->dev);
+       return msm_drv_probe(&pdev->dev, mdp4_kms_init);
 }
 
 static int mdp4_remove(struct platform_device *pdev)
 
                                                          slave_encoder);
 }
 
+static void mdp5_destroy(struct platform_device *pdev);
+
 static void mdp5_kms_destroy(struct msm_kms *kms)
 {
        struct mdp5_kms *mdp5_kms = to_mdp5_kms(to_mdp_kms(kms));
        }
 
        mdp_kms_destroy(&mdp5_kms->base);
+       mdp5_destroy(mdp5_kms->pdev);
 }
 
 #ifdef CONFIG_DEBUG_FS
        return 0;
 }
 
+static int mdp5_init(struct platform_device *pdev, struct drm_device *dev);
+
 static int mdp5_kms_init(struct drm_device *dev)
 {
        struct msm_drm_private *priv = dev->dev_private;
        int irq, i, ret;
        struct device *iommu_dev;
 
+       ret = mdp5_init(to_platform_device(dev->dev), dev);
+
        /* priv->kms would have been populated by the MDP5 driver */
        kms = priv->kms;
        if (!kms)
        u32 major, minor;
        int ret;
 
-       priv->kms_init = mdp5_kms_init;
-
        mdp5_kms = devm_kzalloc(&pdev->dev, sizeof(*mdp5_kms), GFP_KERNEL);
        if (!mdp5_kms) {
                ret = -ENOMEM;
        return ret;
 }
 
-static int mdp5_bind(struct device *dev, struct device *master, void *data)
-{
-       struct msm_drm_private *priv = dev_get_drvdata(master);
-       struct drm_device *ddev = priv->dev;
-       struct platform_device *pdev = to_platform_device(dev);
-
-       DBG("");
-
-       return mdp5_init(pdev, ddev);
-}
-
-static void mdp5_unbind(struct device *dev, struct device *master,
-                       void *data)
-{
-       struct platform_device *pdev = to_platform_device(dev);
-
-       mdp5_destroy(pdev);
-}
-
-static const struct component_ops mdp5_ops = {
-       .bind   = mdp5_bind,
-       .unbind = mdp5_unbind,
-};
-
 static int mdp5_setup_interconnect(struct platform_device *pdev)
 {
-       struct icc_path *path0 = of_icc_get(&pdev->dev, "mdp0-mem");
-       struct icc_path *path1 = of_icc_get(&pdev->dev, "mdp1-mem");
-       struct icc_path *path_rot = of_icc_get(&pdev->dev, "rotator-mem");
+       /* Interconnects are a part of MDSS device tree binding, not the
+        * MDP5 device. */
+       struct device *mdss_dev = pdev->dev.parent;
+       struct icc_path *path0 = of_icc_get(mdss_dev, "mdp0-mem");
+       struct icc_path *path1 = of_icc_get(mdss_dev, "mdp1-mem");
+       struct icc_path *path_rot = of_icc_get(mdss_dev, "rotator-mem");
 
        if (IS_ERR(path0))
                return PTR_ERR(path0);
        if (ret)
                return ret;
 
-       return component_add(&pdev->dev, &mdp5_ops);
+       return msm_drv_probe(&pdev->dev, mdp5_kms_init);
 }
 
 static int mdp5_dev_remove(struct platform_device *pdev)
 {
        DBG("");
-       component_del(&pdev->dev, &mdp5_ops);
+       component_master_del(&pdev->dev, &msm_drm_ops);
        return 0;
 }
 
 
 static const struct dev_pm_ops mdp5_pm_ops = {
        SET_RUNTIME_PM_OPS(mdp5_runtime_suspend, mdp5_runtime_resume, NULL)
+       .prepare = msm_pm_prepare,
+       .complete = msm_pm_complete,
 };
 
-const struct of_device_id mdp5_dt_match[] = {
+static const struct of_device_id mdp5_dt_match[] = {
        { .compatible = "qcom,mdp5", },
        /* to support downstream DT files */
        { .compatible = "qcom,mdss_mdp", },
 
  * is no external component that we need to add since LVDS is within MDP4
  * itself.
  */
-static int add_components_mdp(struct device *master_dev, struct device *mdp_dev,
+static int add_components_mdp(struct device *master_dev,
                              struct component_match **matchptr)
 {
-       struct device_node *np = mdp_dev->of_node;
+       struct device_node *np = master_dev->of_node;
        struct device_node *ep_node;
 
        for_each_endpoint_of_node(np, ep_node) {
 
                ret = of_graph_parse_endpoint(ep_node, &ep);
                if (ret) {
-                       DRM_DEV_ERROR(mdp_dev, "unable to parse port endpoint\n");
+                       DRM_DEV_ERROR(master_dev, "unable to parse port endpoint\n");
                        of_node_put(ep_node);
                        return ret;
                }
        .unbind = msm_drm_unbind,
 };
 
-int msm_drv_probe(struct device *master_dev, struct device *mdp_dev)
+int msm_drv_probe(struct device *master_dev,
+       int (*kms_init)(struct drm_device *dev))
 {
+       struct msm_drm_private *priv;
        struct component_match *match = NULL;
        int ret;
 
-       if (mdp_dev) {
-               /* add the MDP component itself */
-               drm_of_component_match_add(master_dev, &match, component_compare_of,
-                               mdp_dev->of_node);
+       priv = devm_kzalloc(master_dev, sizeof(*priv), GFP_KERNEL);
+       if (!priv)
+               return -ENOMEM;
 
-               ret = add_components_mdp(master_dev, mdp_dev, &match);
+       priv->kms_init = kms_init;
+       dev_set_drvdata(master_dev, priv);
+
+       /* Add mdp components if we have KMS. */
+       if (kms_init) {
+               ret = add_components_mdp(master_dev, &match);
                if (ret)
                        return ret;
        }
 
 static int msm_pdev_probe(struct platform_device *pdev)
 {
-       struct msm_drm_private *priv;
-
-       priv = devm_kzalloc(&pdev->dev, sizeof(*priv), GFP_KERNEL);
-       if (!priv)
-               return -ENOMEM;
-
-       platform_set_drvdata(pdev, priv);
-
        return msm_drv_probe(&pdev->dev, NULL);
 }
 
 
        /* subordinate devices, if present: */
        struct platform_device *gpu_pdev;
 
-       /* top level MDSS wrapper device (for MDP5/DPU only) */
-       struct msm_mdss *mdss;
-
        /* possibly this should be in the kms component, but it is
         * shared by both mdp4 and mdp5..
         */
 int msm_pm_prepare(struct device *dev);
 void msm_pm_complete(struct device *dev);
 
-int msm_drv_probe(struct device *master_dev, struct device *mdp_dev);
+int msm_drv_probe(struct device *dev,
+       int (*kms_init)(struct drm_device *dev));
 void msm_drv_shutdown(struct platform_device *pdev);
 
 
 
                msm_atomic_destroy_pending_timer(&kms->pending_timers[i]);
 }
 
-extern const struct of_device_id dpu_dt_match[];
-extern const struct of_device_id mdp5_dt_match[];
-
 #define for_each_crtc_mask(dev, crtc, crtc_mask) \
        drm_for_each_crtc(crtc, dev) \
                for_each_if (drm_crtc_mask(crtc) & (crtc_mask))
 
 
 static int __maybe_unused mdss_runtime_suspend(struct device *dev)
 {
-       struct msm_drm_private *priv = dev_get_drvdata(dev);
+       struct msm_mdss *mdss = dev_get_drvdata(dev);
 
        DBG("");
 
-       return msm_mdss_disable(priv->mdss);
+       return msm_mdss_disable(mdss);
 }
 
 static int __maybe_unused mdss_runtime_resume(struct device *dev)
 {
-       struct msm_drm_private *priv = dev_get_drvdata(dev);
+       struct msm_mdss *mdss = dev_get_drvdata(dev);
 
        DBG("");
 
-       return msm_mdss_enable(priv->mdss);
+       return msm_mdss_enable(mdss);
 }
 
 static int __maybe_unused mdss_pm_suspend(struct device *dev)
 static const struct dev_pm_ops mdss_pm_ops = {
        SET_SYSTEM_SLEEP_PM_OPS(mdss_pm_suspend, mdss_pm_resume)
        SET_RUNTIME_PM_OPS(mdss_runtime_suspend, mdss_runtime_resume, NULL)
-       .prepare = msm_pm_prepare,
-       .complete = msm_pm_complete,
 };
 
-static int find_mdp_node(struct device *dev, void *data)
-{
-#ifdef CONFIG_DRM_MSM_DPU
-       if (of_match_node(dpu_dt_match, dev->of_node))
-               return true;
-#endif
-
-#ifdef CONFIG_DRM_MSM_MDP5
-       if (of_match_node(mdp5_dt_match, dev->of_node))
-               return true;
-#endif
-
-       return false;
-}
-
 static int mdss_probe(struct platform_device *pdev)
 {
        struct msm_mdss *mdss;
-       struct msm_drm_private *priv;
        bool is_mdp5 = of_device_is_compatible(pdev->dev.of_node, "qcom,mdss");
-       struct device *mdp_dev;
        struct device *dev = &pdev->dev;
        int ret;
 
        if (IS_ERR(mdss))
                return PTR_ERR(mdss);
 
-       priv = devm_kzalloc(&pdev->dev, sizeof(*priv), GFP_KERNEL);
-       if (!priv) {
-               ret = -ENOMEM;
-               goto fail;
-       }
-
-       priv->mdss = mdss;
-       platform_set_drvdata(pdev, priv);
+       platform_set_drvdata(pdev, mdss);
 
        /*
         * MDP5/DPU based devices don't have a flat hierarchy. There is a top
        ret = of_platform_populate(dev->of_node, NULL, NULL, dev);
        if (ret) {
                DRM_DEV_ERROR(dev, "failed to populate children devices\n");
-               goto fail;
-       }
-
-       mdp_dev = device_find_child(dev, NULL, find_mdp_node);
-       if (!mdp_dev) {
-               DRM_DEV_ERROR(dev, "failed to find MDSS MDP node\n");
-               of_platform_depopulate(dev);
-               ret = -ENODEV;
-               goto fail;
+               msm_mdss_destroy(mdss);
+               return ret;
        }
 
-       /*
-        * on MDP5 based platforms, the MDSS platform device is the component
-        * that adds MDP5 and other display interface components to
-        * itself.
-        */
-       ret = msm_drv_probe(dev, mdp_dev);
-       put_device(mdp_dev);
-       if (ret)
-               goto fail;
-
        return 0;
-
-fail:
-       of_platform_depopulate(dev);
-       msm_mdss_destroy(priv->mdss);
-
-       return ret;
 }
 
 static int mdss_remove(struct platform_device *pdev)
 {
-       struct msm_drm_private *priv = platform_get_drvdata(pdev);
-       struct msm_mdss *mdss = priv->mdss;
+       struct msm_mdss *mdss = platform_get_drvdata(pdev);
 
-       component_master_del(&pdev->dev, &msm_drm_ops);
        of_platform_depopulate(&pdev->dev);
 
        msm_mdss_destroy(mdss);