1 files changed, 5 insertions, 1 deletions

diff --git a/mm/slab.h b/mm/slab.h
index 6885e1192ec5..073362816acc 100644
--- a/mm/slab.h
+++ b/mm/slab.h
@@ -43,6 +43,7 @@ struct kmem_cache {
 #include <linux/kasan.h>
 #include <linux/kmemleak.h>
 #include <linux/random.h>
+#include <linux/sched/mm.h>
 
 /*
  * State of the slab allocator.
@@ -412,7 +413,10 @@ static inline struct kmem_cache *slab_pre_alloc_hook(struct kmem_cache *s,
                                                      gfp_t flags)
 {
         flags &= gfp_allowed_mask;
-        lockdep_trace_alloc(flags);
+
+        fs_reclaim_acquire(flags);
+        fs_reclaim_release(flags);
+
         might_sleep_if(gfpflags_allow_blocking(flags));
 
         if (should_failslab(s, flags))