2 files changed, 4 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/common/linux/driver_common.c b/drivers/gpu/nvgpu/common/linux/driver_common.c
index b4e6a02c..7c4645a8 100644
--- a/drivers/gpu/nvgpu/common/linux/driver_common.c
+++ b/drivers/gpu/nvgpu/common/linux/driver_common.c
@@ -163,6 +163,8 @@ static void nvgpu_init_mm_vars(struct gk20a *g)
                             platform->unified_memory);
         __nvgpu_set_enabled(g, NVGPU_MM_UNIFY_ADDRESS_SPACES,
                             platform->unify_address_spaces);
+        __nvgpu_set_enabled(g, NVGPU_MM_CE_TSG_REQUIRED,
+                            platform->tsg_required);
 
         nvgpu_mutex_init(&g->mm.tlb_lock);
         nvgpu_mutex_init(&g->mm.priv_lock);
diff --git a/drivers/gpu/nvgpu/common/linux/pci.c b/drivers/gpu/nvgpu/common/linux/pci.c
index 92764c21..401080ed 100644
--- a/drivers/gpu/nvgpu/common/linux/pci.c
+++ b/drivers/gpu/nvgpu/common/linux/pci.c
@@ -240,6 +240,7 @@ static struct gk20a_platform nvgpu_pci_device[] = {
         .vbios_min_version = 0x88001e00,
         .hardcode_sw_threshold = false,
         .run_preos = true,
+        .tsg_required = true,
         },
         { /* DEVICE=PG503 SKU 200 ES */
         /* ptimer src frequency in hz */
@@ -274,6 +275,7 @@ static struct gk20a_platform nvgpu_pci_device[] = {
         .vbios_min_version = 0x88001e00,
         .hardcode_sw_threshold = false,
         .run_preos = true,
+        .tsg_required = true,
         }
 };