unsigned int cpu;
 
                cpu_clear(smp_processor_id(), mask);
-               for_each_online_cpu(cpu)
+               for_each_cpu_mask(cpu, mask)
                        if (cpu_context(cpu, mm))
                                cpu_context(cpu, mm) = 0;
        }
                unsigned int cpu;
 
                cpu_clear(smp_processor_id(), mask);
-               for_each_online_cpu(cpu)
+               for_each_cpu_mask(cpu, mask)
                        if (cpu_context(cpu, mm))
                                cpu_context(cpu, mm) = 0;
        }
                unsigned int cpu;
 
                cpu_clear(smp_processor_id(), mask);
-               for_each_online_cpu(cpu)
+               for_each_cpu_mask(cpu, mask)
                        if (cpu_context(cpu, vma->vm_mm))
                                cpu_context(cpu, vma->vm_mm) = 0;
        }