rk_obj->dma_addr = rk_obj->mm.start;
 
-       ret = iommu_map_sg(private->domain, rk_obj->dma_addr, rk_obj->sgt->sgl,
-                          rk_obj->sgt->nents, prot);
+       ret = iommu_map_sgtable(private->domain, rk_obj->dma_addr, rk_obj->sgt,
+                               prot);
        if (ret < rk_obj->base.size) {
                DRM_ERROR("failed to map buffer: size=%zd request_size=%zd\n",
                          ret, rk_obj->base.size);
         * TODO: Replace this by drm_clflush_sg() once it can be implemented
         * without relying on symbols that are not exported.
         */
-       for_each_sg(rk_obj->sgt->sgl, s, rk_obj->sgt->nents, i)
+       for_each_sgtable_sg(rk_obj->sgt, s, i)
                sg_dma_address(s) = sg_phys(s);
 
-       dma_sync_sg_for_device(drm->dev, rk_obj->sgt->sgl, rk_obj->sgt->nents,
-                              DMA_TO_DEVICE);
+       dma_sync_sgtable_for_device(drm->dev, rk_obj->sgt, DMA_TO_DEVICE);
 
        return 0;
 
                if (private->domain) {
                        rockchip_gem_iommu_unmap(rk_obj);
                } else {
-                       dma_unmap_sg(drm->dev, rk_obj->sgt->sgl,
-                                    rk_obj->sgt->nents, DMA_BIDIRECTIONAL);
+                       dma_unmap_sgtable(drm->dev, rk_obj->sgt,
+                                         DMA_BIDIRECTIONAL, 0);
                }
                drm_prime_gem_destroy(obj, rk_obj->sgt);
        } else {
                        struct sg_table *sg,
                        struct rockchip_gem_object *rk_obj)
 {
-       int count = dma_map_sg(drm->dev, sg->sgl, sg->nents,
-                              DMA_BIDIRECTIONAL);
-       if (!count)
-               return -EINVAL;
+       int err = dma_map_sgtable(drm->dev, sg, DMA_BIDIRECTIONAL, 0);
+       if (err)
+               return err;
 
        if (drm_prime_get_contiguous_size(sg) < attach->dmabuf->size) {
                DRM_ERROR("failed to map sg_table to contiguous linear address.\n");
-               dma_unmap_sg(drm->dev, sg->sgl, sg->nents,
-                            DMA_BIDIRECTIONAL);
+               dma_unmap_sgtable(drm->dev, sg, DMA_BIDIRECTIONAL, 0);
                return -EINVAL;
        }