if (ret < 0)
                return;
 
-       if (WARN_ON(job_read(pfdev, JS_COMMAND_NEXT(js))))
-               goto end;
+       if (WARN_ON(job_read(pfdev, JS_COMMAND_NEXT(js)))) {
+               pm_runtime_put_sync_autosuspend(pfdev->dev);
+               return;
+       }
 
        cfg = panfrost_mmu_as_get(pfdev, &job->file_priv->mmu);
 
        job_write(pfdev, JS_COMMAND_NEXT(js), JS_COMMAND_START);
 
        spin_unlock_irqrestore(&pfdev->hwaccess_lock, flags);
-
-end:
-       pm_runtime_mark_last_busy(pfdev->dev);
-       pm_runtime_put_autosuspend(pfdev->dev);
 }
 
 static void panfrost_acquire_object_fences(struct drm_gem_object **bos,
        struct panfrost_job *job = to_panfrost_job(sched_job);
        struct panfrost_device *pfdev = job->pfdev;
        int js = panfrost_job_get_slot(job);
+       unsigned long flags;
        int i;
 
        /*
        if (sched_job)
                drm_sched_increase_karma(sched_job);
 
+       spin_lock_irqsave(&pfdev->js->job_lock, flags);
+       for (i = 0; i < NUM_JOB_SLOTS; i++) {
+               if (pfdev->jobs[i]) {
+                       pm_runtime_put_noidle(pfdev->dev);
+                       pfdev->jobs[i] = NULL;
+               }
+       }
+       spin_unlock_irqrestore(&pfdev->js->job_lock, flags);
+
        /* panfrost_core_dump(pfdev); */
 
        panfrost_devfreq_record_transition(pfdev, js);
                }
 
                if (status & JOB_INT_MASK_DONE(j)) {
-                       struct panfrost_job *job = pfdev->jobs[j];
+                       struct panfrost_job *job;
+
+                       spin_lock(&pfdev->js->job_lock);
+                       job = pfdev->jobs[j];
+                       /* Only NULL if job timeout occurred */
+                       if (job) {
+                               pfdev->jobs[j] = NULL;
+
+                               panfrost_mmu_as_put(pfdev, &job->file_priv->mmu);
+                               panfrost_devfreq_record_transition(pfdev, j);
 
-                       pfdev->jobs[j] = NULL;
-                       panfrost_mmu_as_put(pfdev, &job->file_priv->mmu);
-                       panfrost_devfreq_record_transition(pfdev, j);
-                       dma_fence_signal(job->done_fence);
+                               dma_fence_signal_locked(job->done_fence);
+                               pm_runtime_put_autosuspend(pfdev->dev);
+                       }
+                       spin_unlock(&pfdev->js->job_lock);
                }
 
                status &= ~mask;