#include <linux/uaccess.h>
#include <asm/tlb.h>
#include <asm/tlbflush.h>
+#ifdef CONFIG_FINEGRAINED_THP
#include <asm/huge_mm.h>
#include <asm/finegrained_thp.h>
+#else
+#include <asm-generic/huge_mm.h>
+#include <asm-generic/finegrained_thp.h>
+#endif
#include "pgalloc-track.h"
#include "internal.h"