drm/rockchip: fix iommu_map_sg check

[firefly-linux-kernel-4.4.55.git] / drivers / gpu / drm / rockchip / rockchip_drm_gem.c

diff --git a/drivers/gpu/drm/rockchip/rockchip_drm_gem.c b/drivers/gpu/drm/rockchip/rockchip_drm_gem.c
index f79238149ff6e632db8fe540ab2b96c0bdebffc0..2af62e3c6d249df86f431ac47ef9a9a7f761e4a0 100644 (file)
--- a/drivers/gpu/drm/rockchip/rockchip_drm_gem.c
+++ b/drivers/gpu/drm/rockchip/rockchip_drm_gem.c
@@ -47,7 +47,7 @@ static int rockchip_gem_iommu_map(struct rockchip_gem_object *rk_obj)
 
        ret = iommu_map_sg(private->domain, rk_obj->dma_addr, rk_obj->sgt->sgl,
                           rk_obj->sgt->nents, prot);
-       if (ret < 0) {
+       if (ret < rk_obj->base.size) {
                DRM_ERROR("failed to map buffer: %zd\n", ret);
                goto err_remove_node;
        }