1 files changed, 5 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/vgpu/gp10b/vgpu_mm_gp10b.c b/drivers/gpu/nvgpu/vgpu/gp10b/vgpu_mm_gp10b.c
index 4b06aea8..1f537d48 100644
--- a/drivers/gpu/nvgpu/vgpu/gp10b/vgpu_mm_gp10b.c
+++ b/drivers/gpu/nvgpu/vgpu/gp10b/vgpu_mm_gp10b.c
@@ -1,7 +1,7 @@
 /*
  * Virtualized GPU Memory Management
  *
- * Copyright (c) 2015-2019, NVIDIA CORPORATION.  All rights reserved.
+ * Copyright (c) 2015-2020, NVIDIA CORPORATION.  All rights reserved.
  *
  * Permission is hereby granted, free of charge, to any person obtaining a
  * copy of this software and associated documentation files (the "Software"),
@@ -179,6 +179,10 @@ u64 vgpu_gp10b_locked_gmmu_map(struct vm_gk20a *vm,
                 p->flags |= TEGRA_VGPU_MAP_IO_COHERENT;
         if (flags & NVGPU_VM_MAP_L3_ALLOC)
                 p->flags |= TEGRA_VGPU_MAP_L3_ALLOC;
+        if (flags & NVGPU_VM_MAP_PLATFORM_ATOMIC) {
+                p->flags |= TEGRA_VGPU_MAP_PLATFORM_ATOMIC;
+        }
+
         p->prot = prot;
         p->ctag_offset = ctag_offset;
         p->clear_ctags = clear_ctags;