unsigned int highest_zoneidx = MAX_NR_ZONES - 1;
        pg_data_t *pgdat = (pg_data_t *)p;
        struct task_struct *tsk = current;
-       const struct cpumask *cpumask = cpumask_of_node(pgdat->node_id);
-
-       if (!cpumask_empty(cpumask))
-               set_cpus_allowed_ptr(tsk, cpumask);
 
        /*
         * Tell the memory management that we're a "memory allocator",
 
        pgdat_kswapd_lock(pgdat);
        if (!pgdat->kswapd) {
-               pgdat->kswapd = kthread_run(kswapd, pgdat, "kswapd%d", nid);
+               pgdat->kswapd = kthread_create_on_node(kswapd, pgdat, nid, "kswapd%d", nid);
                if (IS_ERR(pgdat->kswapd)) {
                        /* failure at boot is fatal */
                        pr_err("Failed to start kswapd on node %d,ret=%ld\n",
                                   nid, PTR_ERR(pgdat->kswapd));
                        BUG_ON(system_state < SYSTEM_RUNNING);
                        pgdat->kswapd = NULL;
+               } else {
+                       wake_up_process(pgdat->kswapd);
                }
        }
        pgdat_kswapd_unlock(pgdat);