drm/amdgpu: PRT support for gfx9 (v3)

Fix PRT handling on gfx9

v2: unify PRT bit for all ASICs
v3: move PRT flag checking in amdgpu_vm_bo_split_mapping()

Signed-off-by: Junwei Zhang <Jerry.Zhang@amd.com>
Acked-by: David Zhou <david1.zhou@amd.com>
Reviewed-by: Christian König <christian.koenig@amd.com>
Signed-off-by: Alex Deucher <alexander.deucher@amd.com>

1 files changed, 6 insertions, 0 deletions

diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c b/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c
index f34d822f92ac..c42a9979d056 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c
@@ -1338,6 +1338,12 @@ static int amdgpu_vm_bo_split_mapping(struct amdgpu_device *adev,
         flags &= ~AMDGPU_PTE_MTYPE_MASK;
         flags |= (mapping->flags & AMDGPU_PTE_MTYPE_MASK);
 
+        if ((mapping->flags & AMDGPU_PTE_PRT) &&
+            (adev->asic_type >= CHIP_VEGA10)) {
+                flags |= AMDGPU_PTE_PRT;
+                flags &= ~AMDGPU_PTE_VALID;
+        }
+
         trace_amdgpu_vm_bo_update(mapping);
 
         pfn = mapping->offset >> PAGE_SHIFT;