diff --git a/drivers/base/regmap/regmap.c b/drivers/base/regmap/regmap.c
index e57b7bc035fc66a59c8c58ceab2dd6bd6e9e5bab..0bd5d0a748a6e63b21b4d6506d82a484d998ee83 100644
--- a/drivers/base/regmap/regmap.c
+++ b/drivers/base/regmap/regmap.c
@@ -16,6 +16,7 @@
 #include <linux/mutex.h>
 #include <linux/err.h>
 #include <linux/rbtree.h>
+#include <linux/sched.h>
 
 #define CREATE_TRACE_POINTS
 #include <trace/events/regmap.h>