unsigned int pitch;
 
        /* memory management */
-       void __iomem *screen_base;
+       struct iosys_map screen_base;
 
        /* modesetting */
        uint32_t formats[8];
 
        drm_atomic_helper_damage_iter_init(&iter, old_plane_state, plane_state);
        drm_atomic_for_each_plane_damage(&iter, &damage) {
-               struct iosys_map dst = IOSYS_MAP_INIT_VADDR(sdev->screen_base);
                struct drm_rect dst_clip = plane_state->dst;
+               struct iosys_map dst = sdev->screen_base;
 
                if (!drm_rect_intersect(&dst_clip, &damage))
                        continue;
 
                iosys_map_incr(&dst, drm_fb_clip_offset(sdev->pitch, sdev->format, &dst_clip));
-               drm_fb_blit(&dst, &sdev->pitch, sdev->format->format, shadow_plane_state->data, fb,
-                           &damage);
+               drm_fb_blit(&dst, &sdev->pitch, sdev->format->format, shadow_plane_state->data,
+                           fb, &damage);
        }
 
        drm_dev_exit(idx);
                return;
 
        /* Clear screen to black if disabled */
-       memset_io(sdev->screen_base, 0, sdev->pitch * sdev->mode.vdisplay);
+       iosys_map_memset(&sdev->screen_base, 0, 0, sdev->pitch * sdev->mode.vdisplay);
 
        drm_dev_exit(idx);
 }
        screen_base = devm_ioremap_wc(&pdev->dev, mem->start, resource_size(mem));
        if (!screen_base)
                return ERR_PTR(-ENOMEM);
-       sdev->screen_base = screen_base;
+
+       iosys_map_set_vaddr_iomem(&sdev->screen_base, screen_base);
 
        /*
         * Modesetting