1 files changed, 2 insertions, 2 deletions

diff --git a/drivers/gpu/drm/msm/msm_gem.c b/drivers/gpu/drm/msm/msm_gem.c
index 38b7f4e9e7ff..51a95da694d8 100644
--- a/drivers/gpu/drm/msm/msm_gem.c
+++ b/drivers/gpu/drm/msm/msm_gem.c
@@ -975,7 +975,7 @@ static struct drm_gem_object *_msm_gem_new(struct drm_device *dev,
 
         size = PAGE_ALIGN(size);
 
-        if (!iommu_present(&platform_bus_type))
+        if (!msm_use_mmu(dev))
                 use_vram = true;
         else if ((flags & (MSM_BO_STOLEN | MSM_BO_SCANOUT)) && priv->vram.size)
                 use_vram = true;
@@ -1052,7 +1052,7 @@ struct drm_gem_object *msm_gem_import(struct drm_device *dev,
         int ret, npages;
 
         /* if we don't have IOMMU, don't bother pretending we can import: */
-        if (!iommu_present(&platform_bus_type)) {
+        if (!msm_use_mmu(dev)) {
                 DRM_DEV_ERROR(dev->dev, "cannot import without IOMMU\n");
                 return ERR_PTR(-EINVAL);
         }