#include <drm/drm_fourcc.h>
 #include <drm/drm_gem_dma_helper.h>
 #include <drm/drm_gem_framebuffer_helper.h>
+#include <drm/drm_managed.h>
 #include <drm/drm_modeset_helper.h>
 #include <drm/drm_module.h>
 #include <drm/drm_of.h>
        int ret = 0, i;
        u32 version, out_depth = 0;
 
-       malidp = devm_kzalloc(dev, sizeof(*malidp), GFP_KERNEL);
-       if (!malidp)
-               return -ENOMEM;
+       malidp = devm_drm_dev_alloc(dev, &malidp_driver, typeof(*malidp), base);
+       if (IS_ERR(malidp))
+               return PTR_ERR(malidp);
+
+       drm = &malidp->base;
 
-       hwdev = devm_kzalloc(dev, sizeof(*hwdev), GFP_KERNEL);
+       hwdev = drmm_kzalloc(drm, sizeof(*hwdev), GFP_KERNEL);
        if (!hwdev)
                return -ENOMEM;
 
        if (ret && ret != -ENODEV)
                return ret;
 
-       drm = drm_dev_alloc(&malidp_driver, dev);
-       if (IS_ERR(drm)) {
-               ret = PTR_ERR(drm);
-               goto alloc_fail;
-       }
-
        drm->dev_private = malidp;
        dev_set_drvdata(dev, drm);
 
                malidp_runtime_pm_suspend(dev);
        drm->dev_private = NULL;
        dev_set_drvdata(dev, NULL);
-       drm_dev_put(drm);
-alloc_fail:
        of_reserved_mem_device_release(dev);
 
        return ret;
                malidp_runtime_pm_suspend(dev);
        drm->dev_private = NULL;
        dev_set_drvdata(dev, NULL);
-       drm_dev_put(drm);
        of_reserved_mem_device_release(dev);
 }