1 files changed, 4 insertions, 4 deletions

diff --git a/drivers/gpu/drm/nouveau/nouveau_bios.c b/drivers/gpu/drm/nouveau/nouveau_bios.c
index b59f348f14fc..7369b5e73649 100644
--- a/drivers/gpu/drm/nouveau/nouveau_bios.c
+++ b/drivers/gpu/drm/nouveau/nouveau_bios.c
@@ -2083,11 +2083,11 @@ peek_fb(struct drm_device *dev, struct io_mapping *fb,
         uint32_t val = 0;
 
         if (off < pci_resource_len(dev->pdev, 1)) {
-                uint32_t __iomem *p = io_mapping_map_atomic_wc(fb, off);
+                uint32_t __iomem *p = io_mapping_map_atomic_wc(fb, off, KM_USER0);
 
                 val = ioread32(p);
 
-                io_mapping_unmap_atomic(p);
+                io_mapping_unmap_atomic(p, KM_USER0);
         }
 
         return val;
@@ -2098,12 +2098,12 @@ poke_fb(struct drm_device *dev, struct io_mapping *fb,
         uint32_t off, uint32_t val)
 {
         if (off < pci_resource_len(dev->pdev, 1)) {
-                uint32_t __iomem *p = io_mapping_map_atomic_wc(fb, off);
+                uint32_t __iomem *p = io_mapping_map_atomic_wc(fb, off, KM_USER0);
 
                 iowrite32(val, p);
                 wmb();
 
-                io_mapping_unmap_atomic(p);
+                io_mapping_unmap_atomic(p, KM_USER0);
         }
 }