switch (args->param) {
        case DRM_VC4_PARAM_V3D_IDENT0:
-               ret = pm_runtime_get_sync(&vc4->v3d->pdev->dev);
-               if (ret < 0)
+               ret = vc4_v3d_pm_get(vc4);
+               if (ret)
                        return ret;
                args->value = V3D_READ(V3D_IDENT0);
-               pm_runtime_mark_last_busy(&vc4->v3d->pdev->dev);
-               pm_runtime_put_autosuspend(&vc4->v3d->pdev->dev);
+               vc4_v3d_pm_put(vc4);
                break;
        case DRM_VC4_PARAM_V3D_IDENT1:
-               ret = pm_runtime_get_sync(&vc4->v3d->pdev->dev);
-               if (ret < 0)
+               ret = vc4_v3d_pm_get(vc4);
+               if (ret)
                        return ret;
                args->value = V3D_READ(V3D_IDENT1);
-               pm_runtime_mark_last_busy(&vc4->v3d->pdev->dev);
-               pm_runtime_put_autosuspend(&vc4->v3d->pdev->dev);
+               vc4_v3d_pm_put(vc4);
                break;
        case DRM_VC4_PARAM_V3D_IDENT2:
-               ret = pm_runtime_get_sync(&vc4->v3d->pdev->dev);
-               if (ret < 0)
+               ret = vc4_v3d_pm_get(vc4);
+               if (ret)
                        return ret;
                args->value = V3D_READ(V3D_IDENT2);
-               pm_runtime_mark_last_busy(&vc4->v3d->pdev->dev);
-               pm_runtime_put_autosuspend(&vc4->v3d->pdev->dev);
+               vc4_v3d_pm_put(vc4);
                break;
        case DRM_VC4_PARAM_SUPPORTS_BRANCHES:
        case DRM_VC4_PARAM_SUPPORTS_ETC1:
 
 int vc4_v3d_debugfs_ident(struct seq_file *m, void *unused);
 int vc4_v3d_debugfs_regs(struct seq_file *m, void *unused);
 int vc4_v3d_get_bin_slot(struct vc4_dev *vc4);
+int vc4_v3d_pm_get(struct vc4_dev *vc4);
+void vc4_v3d_pm_put(struct vc4_dev *vc4);
 
 /* vc4_validate.c */
 int
 
        /* Release the reference we had on the perf monitor. */
        vc4_perfmon_put(exec->perfmon);
 
-       mutex_lock(&vc4->power_lock);
-       if (--vc4->power_refcount == 0) {
-               pm_runtime_mark_last_busy(&vc4->v3d->pdev->dev);
-               pm_runtime_put_autosuspend(&vc4->v3d->pdev->dev);
-       }
-       mutex_unlock(&vc4->power_lock);
+       vc4_v3d_pm_put(vc4);
 
        kfree(exec);
 }
                return -ENOMEM;
        }
 
-       mutex_lock(&vc4->power_lock);
-       if (vc4->power_refcount++ == 0) {
-               ret = pm_runtime_get_sync(&vc4->v3d->pdev->dev);
-               if (ret < 0) {
-                       mutex_unlock(&vc4->power_lock);
-                       vc4->power_refcount--;
-                       kfree(exec);
-                       return ret;
-               }
+       ret = vc4_v3d_pm_get(vc4);
+       if (ret) {
+               kfree(exec);
+               return ret;
        }
-       mutex_unlock(&vc4->power_lock);
 
        exec->args = args;
        INIT_LIST_HEAD(&exec->unref_list);
 
 }
 #endif /* CONFIG_DEBUG_FS */
 
+/**
+ * Wraps pm_runtime_get_sync() in a refcount, so that we can reliably
+ * get the pm_runtime refcount to 0 in vc4_reset().
+ */
+int
+vc4_v3d_pm_get(struct vc4_dev *vc4)
+{
+       mutex_lock(&vc4->power_lock);
+       if (vc4->power_refcount++ == 0) {
+               int ret = pm_runtime_get_sync(&vc4->v3d->pdev->dev);
+
+               if (ret < 0) {
+                       vc4->power_refcount--;
+                       mutex_unlock(&vc4->power_lock);
+                       return ret;
+               }
+       }
+       mutex_unlock(&vc4->power_lock);
+
+       return 0;
+}
+
+void
+vc4_v3d_pm_put(struct vc4_dev *vc4)
+{
+       mutex_lock(&vc4->power_lock);
+       if (--vc4->power_refcount == 0) {
+               pm_runtime_mark_last_busy(&vc4->v3d->pdev->dev);
+               pm_runtime_put_autosuspend(&vc4->v3d->pdev->dev);
+       }
+       mutex_unlock(&vc4->power_lock);
+}
+
 static void vc4_v3d_init_hw(struct drm_device *dev)
 {
        struct vc4_dev *vc4 = to_vc4_dev(dev);