diff --git a/arch/riscv/kernel/probes/kprobes.c b/arch/riscv/kernel/probes/kprobes.c
index 7e2c78e2ca6b0bc93b19e1a7011648d0e4afc2c6..f3fbb0b837fad8e996a70af7d7a0ccc7e50ca65d 100644
--- a/arch/riscv/kernel/probes/kprobes.c
+++ b/arch/riscv/kernel/probes/kprobes.c
@@ -6,6 +6,7 @@
 #include <linux/stop_machine.h>
 #include <asm/ptrace.h>
 #include <linux/uaccess.h>
+#include <linux/vmalloc.h>
 #include <asm/sections.h>
 #include <asm/cacheflush.h>
 #include <asm/bug.h>