1 files changed, 7 insertions, 6 deletions

diff --git a/include/linux/ioprio.h b/include/linux/ioprio.h
index baf29387cab4..2a3bb1bb7433 100644
--- a/include/linux/ioprio.h
+++ b/include/linux/ioprio.h
@@ -2,6 +2,7 @@
 #define IOPRIO_H
 
 #include <linux/sched.h>
+#include <linux/iocontext.h>
 
 /*
  * Gives us 8 prio classes with 13-bits of data for each class
@@ -45,18 +46,18 @@ enum {
  * the cpu scheduler nice value to an io priority
  */
 #define IOPRIO_NORM     (4)
-static inline int task_ioprio(struct task_struct *task)
+static inline int task_ioprio(struct io_context *ioc)
 {
-        if (ioprio_valid(task->ioprio))
-                return IOPRIO_PRIO_DATA(task->ioprio);
+        if (ioprio_valid(ioc->ioprio))
+                return IOPRIO_PRIO_DATA(ioc->ioprio);
 
         return IOPRIO_NORM;
 }
 
-static inline int task_ioprio_class(struct task_struct *task)
+static inline int task_ioprio_class(struct io_context *ioc)
 {
-        if (ioprio_valid(task->ioprio))
-                return IOPRIO_PRIO_CLASS(task->ioprio);
+        if (ioprio_valid(ioc->ioprio))
+                return IOPRIO_PRIO_CLASS(ioc->ioprio);
 
         return IOPRIO_CLASS_BE;
 }