* Copyright (C) 2020-2021 Linaro Ltd.
  */
 
-#include <linux/delay.h>
 #include <linux/interrupt.h>
 #include <linux/io.h>
 #include <linux/iopoll.h>
        return 0;
 }
 
-static int vfe_disable_output(struct vfe_line *line)
+static void vfe_disable_output(struct vfe_line *line)
 {
        struct vfe_device *vfe = to_vfe(line);
        struct vfe_output *output = &line->output;
        unsigned long flags;
        unsigned int i;
-       bool done;
-       int timeout = 0;
-
-       do {
-               spin_lock_irqsave(&vfe->output_lock, flags);
-               done = !output->gen2.active_num;
-               spin_unlock_irqrestore(&vfe->output_lock, flags);
-               usleep_range(10000, 20000);
-
-               if (timeout++ == 100) {
-                       dev_err(vfe->camss->dev, "VFE idle timeout - resetting\n");
-                       vfe_reset(vfe);
-                       output->gen2.active_num = 0;
-                       return 0;
-               }
-       } while (!done);
 
        spin_lock_irqsave(&vfe->output_lock, flags);
        for (i = 0; i < output->wm_num; i++)
                vfe_wm_stop(vfe, output->wm_idx[i]);
+       output->gen2.active_num = 0;
        spin_unlock_irqrestore(&vfe->output_lock, flags);
 
-       return 0;
+       vfe_reset(vfe);
 }
 
 /*