unsigned long vsid;
        pte_t *ptep;
        unsigned hugeshift;
-       const struct cpumask *tmp;
        int rc, user_region = 0;
        int psize, ssize;
 
        }
 
        /* Check CPU locality */
-       tmp = cpumask_of(smp_processor_id());
-       if (user_region && cpumask_equal(mm_cpumask(mm), tmp))
+       if (user_region && mm_is_thread_local(mm))
                flags |= HPTE_LOCAL_UPDATE;
 
 #ifndef CONFIG_PPC_64K_PAGES
 #endif /* CONFIG_PPC_64K_PAGES */
 
        /* Is that local to this CPU ? */
-       if (cpumask_equal(mm_cpumask(mm), cpumask_of(smp_processor_id())))
+       if (mm_is_thread_local(mm))
                update_flags |= HPTE_LOCAL_UPDATE;
 
        /* Hash it in */
 
        batchp = &get_cpu_var(hugepd_freelist_cur);
 
        if (atomic_read(&tlb->mm->mm_users) < 2 ||
-           cpumask_equal(mm_cpumask(tlb->mm),
-                         cpumask_of(smp_processor_id()))) {
+           mm_is_thread_local(tlb->mm)) {
                kmem_cache_free(hugepte_cache, hugepte);
                put_cpu_var(hugepd_freelist_cur);
                return;
 
        unsigned int psize;
        unsigned long vsid;
        unsigned long flags = 0;
-       const struct cpumask *tmp;
 
        /* get the base page size,vsid and segment size */
 #ifdef CONFIG_DEBUG_VM
                ssize = mmu_kernel_ssize;
        }
 
-       tmp = cpumask_of(smp_processor_id());
-       if (cpumask_equal(mm_cpumask(mm), tmp))
+       if (mm_is_thread_local(mm))
                flags |= HPTE_LOCAL_UPDATE;
 
        return flush_hash_hugepage(vsid, addr, pmdp, psize, ssize, flags);
 
  */
 void __flush_tlb_pending(struct ppc64_tlb_batch *batch)
 {
-       const struct cpumask *tmp;
-       int i, local = 0;
+       int i, local;
 
        i = batch->index;
-       tmp = cpumask_of(smp_processor_id());
-       if (cpumask_equal(mm_cpumask(batch->mm), tmp))
-               local = 1;
+       local = mm_is_thread_local(batch->mm);
        if (i == 1)
                flush_hash_page(batch->vpn[0], batch->pte[0],
                                batch->psize, batch->ssize, local);