struct scatterlist *sg;
        int i;
 
-       for_each_sg(data->sg, sg, data->sg_len, i) {
-               void *buf = kmap_atomic(sg_page(sg) + sg->offset);
-               buffer_swap32(buf, sg->length);
-               kunmap_atomic(buf);
-       }
+       for_each_sg(data->sg, sg, data->sg_len, i)
+               buffer_swap32(sg_virt(sg), sg->length);
 }
 #else
 static inline void mxcmci_swap_buffers(struct mmc_data *data) {}
 {
        struct mmc_data *data = host->req->data;
        struct scatterlist *sg;
-       void *buf;
        int stat, i;
 
        host->data = data;
 
        if (data->flags & MMC_DATA_READ) {
                for_each_sg(data->sg, sg, data->sg_len, i) {
-                       buf = kmap_atomic(sg_page(sg) + sg->offset);
-                       stat = mxcmci_pull(host, buf, sg->length);
-                       kunmap(buf);
+                       stat = mxcmci_pull(host, sg_virt(sg), sg->length);
                        if (stat)
                                return stat;
                        host->datasize += sg->length;
                }
        } else {
                for_each_sg(data->sg, sg, data->sg_len, i) {
-                       buf = kmap_atomic(sg_page(sg) + sg->offset);
-                       stat = mxcmci_push(host, buf, sg->length);
-                       kunmap(buf);
+                       stat = mxcmci_push(host, sg_virt(sg), sg->length);
                        if (stat)
                                return stat;
                        host->datasize += sg->length;