1 files changed, 1 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/os/linux/vgpu/vgpu_linux.c b/drivers/gpu/nvgpu/os/linux/vgpu/vgpu_linux.c
index 83191316..33c067c6 100644
--- a/drivers/gpu/nvgpu/os/linux/vgpu/vgpu_linux.c
+++ b/drivers/gpu/nvgpu/os/linux/vgpu/vgpu_linux.c
@@ -439,8 +439,7 @@ int vgpu_probe(struct platform_device *pdev)
         gk20a_init_gr(gk20a);
 
         nvgpu_log_info(gk20a, "total ram pages : %lu", totalram_pages);
-        gk20a->gr.max_comptag_mem = totalram_pages
-                                 >> (10 - (PAGE_SHIFT - 10));
+        gk20a->gr.max_comptag_mem = totalram_size_in_mb;
 
         nvgpu_ref_init(&gk20a->refcount);