hpage_offset /= 4;
 
        get_page(hpage);
-       page = kmap_atomic(hpage, KM_USER0);
+       page = kmap_atomic(hpage);
 
        /* patch dcbz into reserved instruction, so we trap */
        for (i=hpage_offset; i < hpage_offset + (HW_PAGE_SIZE / 4); i++)
                if ((page[i] & 0xff0007ff) == INS_DCBZ)
                        page[i] &= 0xfffffff7;
 
-       kunmap_atomic(page, KM_USER0);
+       kunmap_atomic(page);
        put_page(hpage);
 }
 
 
        local_irq_save(flags);
 
        do {
-               start = (unsigned long)kmap_atomic(page + seg_nr,
-                               KM_PPC_SYNC_PAGE) + seg_offset;
+               start = (unsigned long)kmap_atomic(page + seg_nr) + seg_offset;
 
                /* Sync this buffer segment */
                __dma_sync((void *)start, seg_size, direction);
-               kunmap_atomic((void *)start, KM_PPC_SYNC_PAGE);
+               kunmap_atomic((void *)start);
                seg_nr++;
 
                /* Calculate next buffer segment size */
 
                if (!PageHighMem(page)) {
                        __flush_dcache_icache(page_address(page+i));
                } else {
-                       start = kmap_atomic(page+i, KM_PPC_SYNC_ICACHE);
+                       start = kmap_atomic(page+i);
                        __flush_dcache_icache(start);
-                       kunmap_atomic(start, KM_PPC_SYNC_ICACHE);
+                       kunmap_atomic(start);
                }
        }
 }
 
 #endif
 #ifdef CONFIG_BOOKE
        {
-               void *start = kmap_atomic(page, KM_PPC_SYNC_ICACHE);
+               void *start = kmap_atomic(page);
                __flush_dcache_icache(start);
-               kunmap_atomic(start, KM_PPC_SYNC_ICACHE);
+               kunmap_atomic(start);
        }
 #elif defined(CONFIG_8xx) || defined(CONFIG_PPC64)
        /* On 8xx there is no need to kmap since highmem is not supported */