1 files changed, 2 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/common/linux/vgpu/gp10b/vgpu_mm_gp10b.c b/drivers/gpu/nvgpu/common/linux/vgpu/gp10b/vgpu_mm_gp10b.c
index 72214d3c..161a69e9 100644
--- a/drivers/gpu/nvgpu/common/linux/vgpu/gp10b/vgpu_mm_gp10b.c
+++ b/drivers/gpu/nvgpu/common/linux/vgpu/gp10b/vgpu_mm_gp10b.c
@@ -23,10 +23,10 @@
 #include "gk20a/mm_gk20a.h"
 
 #include <nvgpu/bug.h>
+#include <nvgpu/dma.h>
 
 int vgpu_gp10b_init_mm_setup_hw(struct gk20a *g)
 {
-        g->mm.bypass_smmu = true;
         g->mm.disable_bigpage = true;
         return 0;
 }
@@ -77,7 +77,7 @@ u64 vgpu_gp10b_locked_gmmu_map(struct vm_gk20a *vm,
 
         /* FIXME: add support for sparse mappings */
 
-        if (WARN_ON(!sgt) || WARN_ON(!g->mm.bypass_smmu))
+        if (WARN_ON(!sgt) || WARN_ON(nvgpu_iommuable(g)))
                 return 0;
 
         if (space_to_skip & (page_size - 1))