#include <linux/module.h>
 #include <linux/of_device.h>
 #include <linux/of.h>
+#include <linux/pm_runtime.h>
 #include <media/v4l2-event.h>
 #include <media/v4l2-mem2mem.h>
 #include <media/videobuf2-dma-contig.h>
                return ret;
        }
 
-       ret = mtk_vcodec_init_dec_pm(pdev, &dev->pm);
+       ret = mtk_vcodec_init_dec_clk(pdev, &dev->pm);
        if (ret < 0) {
                dev_err(&pdev->dev, "failed to get mt vcodec clock source");
                return ret;
        }
 
+       pm_runtime_enable(&pdev->dev);
        return 0;
 }
 
        if (IS_VDEC_LAT_ARCH(dev->vdec_pdata->hw_arch))
                destroy_workqueue(dev->core_workqueue);
 err_res:
-       mtk_vcodec_release_dec_pm(&dev->pm);
+       pm_runtime_disable(dev->pm.dev);
+       put_device(dev->pm.larbvdec);
 err_dec_pm:
        mtk_vcodec_fw_release(dev->fw_handler);
        return ret;
                video_unregister_device(dev->vfd_dec);
 
        v4l2_device_unregister(&dev->v4l2_dev);
-       mtk_vcodec_release_dec_pm(&dev->pm);
+       pm_runtime_disable(dev->pm.dev);
+       put_device(dev->pm.larbvdec);
        mtk_vcodec_fw_release(dev->fw_handler);
        return 0;
 }
 
 #include <linux/module.h>
 #include <linux/of.h>
 #include <linux/of_device.h>
+#include <linux/pm_runtime.h>
 #include <linux/slab.h>
 
 #include "mtk_vcodec_drv.h"
                return -ENOMEM;
 
        subdev_dev->plat_dev = pdev;
-       ret = mtk_vcodec_init_dec_pm(pdev, &subdev_dev->pm);
+       ret = mtk_vcodec_init_dec_clk(pdev, &subdev_dev->pm);
        if (ret)
                return ret;
+       pm_runtime_enable(&pdev->dev);
 
        of_id = of_match_device(mtk_vdec_hw_match, dev);
        if (!of_id) {
        platform_set_drvdata(pdev, subdev_dev);
        return 0;
 err:
-       mtk_vcodec_release_dec_pm(&subdev_dev->pm);
+       pm_runtime_disable(subdev_dev->pm.dev);
+       put_device(subdev_dev->pm.larbvdec);
        return ret;
 }
 
 
 #include "mtk_vcodec_dec_pm.h"
 #include "mtk_vcodec_util.h"
 
-int mtk_vcodec_init_dec_pm(struct platform_device *pdev, struct mtk_vcodec_pm *pm)
+int mtk_vcodec_init_dec_clk(struct platform_device *pdev, struct mtk_vcodec_pm *pm)
 {
        struct device_node *node;
        struct platform_device *larb_pdev;
                }
        }
 
-       pm_runtime_enable(&pdev->dev);
        return 0;
 put_device:
        put_device(pm->larbvdec);
        return ret;
 }
-EXPORT_SYMBOL_GPL(mtk_vcodec_init_dec_pm);
-
-void mtk_vcodec_release_dec_pm(struct mtk_vcodec_pm *pm)
-{
-       pm_runtime_disable(pm->dev);
-       put_device(pm->larbvdec);
-}
-EXPORT_SYMBOL_GPL(mtk_vcodec_release_dec_pm);
+EXPORT_SYMBOL_GPL(mtk_vcodec_init_dec_clk);
 
 int mtk_vcodec_dec_pw_on(struct mtk_vcodec_dev *vdec_dev, int hw_idx)
 {
 
 
 #include "mtk_vcodec_drv.h"
 
-int mtk_vcodec_init_dec_pm(struct platform_device *pdev, struct mtk_vcodec_pm *pm);
-void mtk_vcodec_release_dec_pm(struct mtk_vcodec_pm *pm);
+int mtk_vcodec_init_dec_clk(struct platform_device *pdev, struct mtk_vcodec_pm *pm);
 
 int mtk_vcodec_dec_pw_on(struct mtk_vcodec_dev *vdec_dev, int hw_idx);
 void mtk_vcodec_dec_pw_off(struct mtk_vcodec_dev *vdec_dev, int hw_idx);