if (unlikely(page == NULL))
                return;
 
-       page_virtual = kmap_atomic(page, KM_USER0);
+       page_virtual = kmap_atomic(page);
        for (i = 0; i < PAGE_SIZE; i += boot_cpu_data.x86_clflush_size)
                clflush(page_virtual + i);
-       kunmap_atomic(page_virtual, KM_USER0);
+       kunmap_atomic(page_virtual);
 }
 
 static void drm_cache_flush_clflush(struct page *pages[],
                if (unlikely(page == NULL))
                        continue;
 
-               page_virtual = kmap_atomic(page, KM_USER0);
+               page_virtual = kmap_atomic(page);
                flush_dcache_range((unsigned long)page_virtual,
                                   (unsigned long)page_virtual + PAGE_SIZE);
-               kunmap_atomic(page_virtual, KM_USER0);
+               kunmap_atomic(page_virtual);
        }
 #else
        printk(KERN_ERR "Architecture has no drm_cache.c support\n");
 
                        goto out_err;
 
                preempt_disable();
-               from_virtual = kmap_atomic(from_page, KM_USER0);
-               to_virtual = kmap_atomic(to_page, KM_USER1);
+               from_virtual = kmap_atomic(from_page);
+               to_virtual = kmap_atomic(to_page);
                memcpy(to_virtual, from_virtual, PAGE_SIZE);
-               kunmap_atomic(to_virtual, KM_USER1);
-               kunmap_atomic(from_virtual, KM_USER0);
+               kunmap_atomic(to_virtual);
+               kunmap_atomic(from_virtual);
                preempt_enable();
                page_cache_release(from_page);
        }
                        goto out_err;
                }
                preempt_disable();
-               from_virtual = kmap_atomic(from_page, KM_USER0);
-               to_virtual = kmap_atomic(to_page, KM_USER1);
+               from_virtual = kmap_atomic(from_page);
+               to_virtual = kmap_atomic(to_page);
                memcpy(to_virtual, from_virtual, PAGE_SIZE);
-               kunmap_atomic(to_virtual, KM_USER1);
-               kunmap_atomic(from_virtual, KM_USER0);
+               kunmap_atomic(to_virtual);
+               kunmap_atomic(from_virtual);
                preempt_enable();
                set_page_dirty(to_page);
                mark_page_accessed(to_page);
 
 
                if (likely(page_virtual != NULL)) {
                        desc_virtual->ppn = page_to_pfn(page);
-                       kunmap_atomic(page_virtual, KM_USER0);
+                       kunmap_atomic(page_virtual);
                }
 
-               page_virtual = kmap_atomic(page, KM_USER0);
+               page_virtual = kmap_atomic(page);
                desc_virtual = page_virtual - 1;
                prev_pfn = ~(0UL);
 
        }
 
        if (likely(page_virtual != NULL))
-               kunmap_atomic(page_virtual, KM_USER0);
+               kunmap_atomic(page_virtual);
 
        return 0;
 out_err: