#include <linux/proc_fs.h>
 #include <linux/seq_file.h>
 #include <linux/kasan.h>
+#include <linux/node.h>
 #include <linux/kmsan.h>
 #include <linux/cpu.h>
 #include <linux/cpuset.h>
        return __kmem_cache_do_shrink(s);
 }
 
-static int slab_mem_going_offline_callback(void *arg)
+static int slab_mem_going_offline_callback(void)
 {
        struct kmem_cache *s;
 
        return 0;
 }
 
-static int slab_mem_going_online_callback(void *arg)
+static int slab_mem_going_online_callback(int nid)
 {
        struct kmem_cache_node *n;
        struct kmem_cache *s;
-       struct memory_notify *marg = arg;
-       int nid = marg->status_change_nid;
        int ret = 0;
 
-       /*
-        * If the node's memory is already available, then kmem_cache_node is
-        * already created. Nothing to do.
-        */
-       if (nid < 0)
-               return 0;
-
        /*
         * We are bringing a node online. No memory is available yet. We must
         * allocate a kmem_cache_node structure in order to bring the node
 static int slab_memory_callback(struct notifier_block *self,
                                unsigned long action, void *arg)
 {
+       struct node_notify *nn = arg;
+       int nid = nn->nid;
        int ret = 0;
 
        switch (action) {
-       case MEM_GOING_ONLINE:
-               ret = slab_mem_going_online_callback(arg);
-               break;
-       case MEM_GOING_OFFLINE:
-               ret = slab_mem_going_offline_callback(arg);
+       case NODE_ADDING_FIRST_MEMORY:
+               ret = slab_mem_going_online_callback(nid);
                break;
-       case MEM_ONLINE:
-       case MEM_CANCEL_OFFLINE:
+       case NODE_REMOVING_LAST_MEMORY:
+               ret = slab_mem_going_offline_callback();
                break;
        }
        if (ret)
                        sizeof(struct kmem_cache_node),
                        SLAB_HWCACHE_ALIGN | SLAB_NO_OBJ_EXT, 0, 0);
 
-       hotplug_memory_notifier(slab_memory_callback, SLAB_CALLBACK_PRI);
+       hotplug_node_notifier(slab_memory_callback, SLAB_CALLBACK_PRI);
 
        /* Able to allocate the per node structures */
        slab_state = PARTIAL;