This function is entirely self-contained, so can be converted from page
to slab.
Signed-off-by: Matthew Wilcox (Oracle) <willy@infradead.org>
Signed-off-by: Vlastimil Babka <vbabka@suse.cz>
Acked-by: Johannes Weiner <hannes@cmpxchg.org>
Reviewed-by: Roman Gushchin <guro@fb.com>
Reviewed-by: Hyeonggon Yoo <42.hyeyoo@gmail.com>
 
 static inline struct kmem_cache *virt_to_cache(const void *obj)
 {
-       struct page *page;
+       struct slab *slab;
 
-       page = virt_to_head_page(obj);
-       if (WARN_ONCE(!PageSlab(page), "%s: Object is not a Slab page!\n",
+       slab = virt_to_slab(obj);
+       if (WARN_ONCE(!slab, "%s: Object is not a Slab page!\n",
                                        __func__))
                return NULL;
-       return page->slab_cache;
+       return slab->slab_cache;
 }
 
 static __always_inline void account_slab(struct slab *slab, int order,