1 files changed, 6 insertions, 1 deletions
diff --git a/rust/helpers/task.c b/rust/helpers/task.c
index 7d66487db831..2c85bbc2727e 100644
--- a/rust/helpers/task.c
+++ b/rust/helpers/task.c
@@ -1,8 +1,13 @@
 // SPDX-License-Identifier: GPL-2.0
 
-#include <linux/export.h>
+#include <linux/kernel.h>
 #include <linux/sched/task.h>
 
+void rust_helper_might_resched(void)
+{
+	might_resched();
+}
+
 struct task_struct *rust_helper_get_current(void)
 {
 	return current;