#include <asm/tlbflush.h>
 
 int
-reserve_io_memtype_wc(u64 base, unsigned long size, pgprot_t *prot);
-
-void
-free_io_memtype(u64 base, unsigned long size);
+is_io_mapping_possible(resource_size_t base, unsigned long size);
 
 void *
 iomap_atomic_prot_pfn(unsigned long pfn, enum km_type type, pgprot_t prot);
 
 #include <linux/module.h>
 
 #ifdef CONFIG_X86_PAE
-static int
+int
 is_io_mapping_possible(resource_size_t base, unsigned long size)
 {
        return 1;
 }
 #else
-static int
+int
 is_io_mapping_possible(resource_size_t base, unsigned long size)
 {
        /* There is no way to map greater than 1 << 32 address without PAE */
 }
 #endif
 
-int
-reserve_io_memtype_wc(u64 base, unsigned long size, pgprot_t *prot)
-{
-       unsigned long ret_flag;
-
-       if (!is_io_mapping_possible(base, size))
-               goto out_err;
-
-       if (!pat_enabled) {
-               *prot = pgprot_noncached(PAGE_KERNEL);
-               return 0;
-       }
-
-       if (reserve_memtype(base, base + size, _PAGE_CACHE_WC, &ret_flag))
-               goto out_err;
-
-       if (ret_flag == _PAGE_CACHE_WB)
-               goto out_free;
-
-       if (kernel_map_sync_memtype(base, size, ret_flag))
-               goto out_free;
-
-       *prot = __pgprot(__PAGE_KERNEL | ret_flag);
-       return 0;
-
-out_free:
-       free_memtype(base, base + size);
-out_err:
-       return -EINVAL;
-}
-EXPORT_SYMBOL_GPL(reserve_io_memtype_wc);
-
-void
-free_io_memtype(u64 base, unsigned long size)
-{
-       if (pat_enabled)
-               free_memtype(base, base + size);
-}
-EXPORT_SYMBOL_GPL(free_io_memtype);
-
 /* Map 'pfn' using fixed map 'type' and protections 'prot'
  */
 void *
 
 io_mapping_create_wc(resource_size_t base, unsigned long size)
 {
        struct io_mapping *iomap;
-       pgprot_t prot;
 
-       if (!reserve_io_memtype_wc(base, size, &prot))
+       if (!is_io_mapping_possible(base, size))
                return NULL;
 
        iomap = kmalloc(sizeof(*iomap), GFP_KERNEL);
 
        iomap->base = base;
        iomap->size = size;
-       iomap->prot = prot;
+       iomap->prot = pgprot_writecombine(__pgprot(__PAGE_KERNEL));
        return iomap;
 }
 
 static inline void
 io_mapping_free(struct io_mapping *mapping)
 {
-       free_io_memtype(mapping->base, mapping->size);
        kfree(mapping);
 }