1 files changed, 3 insertions, 0 deletions

diff --git a/drivers/staging/comedi/comedi_fops.c b/drivers/staging/comedi/comedi_fops.c
index ca11be21f64b..34ca7823255d 100644
--- a/drivers/staging/comedi/comedi_fops.c
+++ b/drivers/staging/comedi/comedi_fops.c
@@ -2396,6 +2396,7 @@ static ssize_t comedi_write(struct file *file, const char __user *buf,
                         continue;
                 }
 
+                set_current_state(TASK_RUNNING);
                 wp = async->buf_write_ptr;
                 n1 = min(n, async->prealloc_bufsz - wp);
                 n2 = n - n1;
@@ -2528,6 +2529,8 @@ static ssize_t comedi_read(struct file *file, char __user *buf, size_t nbytes,
                         }
                         continue;
                 }
+
+                set_current_state(TASK_RUNNING);
                 rp = async->buf_read_ptr;
                 n1 = min(n, async->prealloc_bufsz - rp);
                 n2 = n - n1;